AWS Neuron ML Compiler Engineer

Accelerate AI workloads as an ML Compiler Engineer at Amazon Web Services (AWS) in the Annapurna Labs team. Design high-performance kernels and optimize machine learning operations using Neuron architecture.

Join the AWS team to contribute to AWS Neuron, enhancing deep learning and GenAI workloads. This role requires expertise in designing compute kernels and conducting performance analyses. You will directly collaborate with customers to optimize their models on AWS accelerators, pushing the frontier of AI capabilities.

Key Responsibilities:

• Design high-performance compute kernels for machine learning operations
• Analyze and optimize kernel performance on Neuron hardware
• Conduct performance analysis using profiling tools
• Implement advanced compiler optimizations like fusion and tiling
• Collaborate cross-functionally to innovate kernel optimization techniques

Requirements

• 3+ years of professional software development experience
• 2+ years in design or architecture roles
• Proficiency in at least one programming language
• Bachelor's degree in computer science preferred
• Strong understanding of coding standards and testing processes

Bring your skills in kernel design and optimization to AWS, driving innovation in machine learning.