AWS Senior ML Kernel Optimization Engineer

Join Amazon's Annapurna Labs as a Senior ML Kernel Optimization Engineer, focusing on enhancing performance for machine learning accelerators. Drive innovation in deep learning and GenAI solutions today.

This role is crucial for developing high-performance kernels, collaborating with cross-functional teams to optimize AWS's Neuron SDK. You will employ your expertise in low-level programming and machine learning to solve complex optimization challenges, ensuring peak performance for diverse workloads.

Key Responsibilities:

• Develop high-performance compute kernels for machine learning

• Optimize kernel performance through detailed analysis

• Utilize profiling tools for performance bottleneck resolution

• Implement advanced compiler enhancements and optimizations

• Collaborate with teams to innovate kernel optimization techniques

Requirements:

• 5+ years in software development roles

• 5+ years of experience in programming languages

• Experience in system design and architecture leadership

• Proven mentoring skills or experience leading teams

• Familiarity with accelerator architectures for machine learning

Drive the future of AI and machine learning at Amazon, delivering cutting-edge solutions for global customers.
#J-18808-Ljbffr