Remote Senior AI GPU Compiler Engineer Job in Cambridge, UK AMD

Who you are

• 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

Desirables

• 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

What the job involves

• 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