Senior AI GPU Compiler Engineer
Skills
Back-End Web DevelopmentComputer EngineeringComputer ScienceSoftware DevelopmentCompilersGraphsHigh Performance Computing
What the job involves
The main requirements, responsibilities and hiring steps.
Requirements
- Strong object-oriented programming background C/C++ preferred
- Proficient in crafting high performance GPU kernels
- Prior experience with compilers dataflow graphs
- Familiarity with modern source control systems such as GitHub
- Excellent verbal communication and writing skills
Nice to have
- Proven expertise in optimizing high performance GPU kernels
- Prior hands-on experience using AI accelerators
- Contributions in LLVM or MLIR projects
- Previous exposure to machine learning models
- Collaborative team player with experience in dispersed teams
Day to day
- Immerse in the open-source AI compiler and runtime stack
- Gain understanding of compiling Machine Learning networks for AMD graphics cards
- Engage with MLIR dialects, execution models, and code generation
- Contribute to developing components within a sophisticated software tool chain in C++
- Explore and contribute to AI compilation technology
- Implement and optimize GPU kernels using compiler technology
- Conduct benchmarking on AI model workloads (vision, speech recognition, language processing, generative models)
- Collaborate with AMD’s architecture specialists to enhance future products
- Stay updated with trends in algorithms, architecture, software, and hardware
- Participate in new GPU bring-ups for AI software
- Debug and resolve existing issues
- Research efficient alternatives for task completion
- Cultivate technical relationships with peers and partners