Senior ML Kernel Optimization Engineer

Amazon Web Services (AWS); Cupertino (CA); On-site; USD 193,300 - 261,500; Full time

Job summary

Annapurna Labs (U.S.) Inc. in Cupertino, CA is seeking a kernel engineer to design high-performance ML compute kernels for Neuron hardware and collaborate across compiler, runtime, and hardware teams to accelerate ML workloads on AWS accelerators.

You will optimize kernel performance across Neuron generations, perform profiling, and implement fusion, tiling, and scheduling while working with customers to tailor ML models for AWS accelerators.

Qualifications

• 5+ years of non-internship professional software development experience.
• 0
• 5+ years of programming with at least one software language.
• 5+ years leading design or architecture of new or existing systems.

Responsibilities

• Design and implement high-performance compute kernels for ML operations leveraging Neuron architecture.
• Analyze and optimize kernel-level performance across multiple Neuron generations.
• Implement compiler optimizations such as fusion, tiling, and scheduling.
• Work with customers to enable and optimize their ML models on AWS accelerators.

Skills

Software development

Programming

Mentorship

Education

Bachelor's degree in computer science or equivalent

Tools

LLVM/MLIR backend development

Job description

Annapurna Labs (U.S.) Inc. in Cupertino, CA is seeking a kernel engineer to design high-performance ML compute kernels for Neuron hardware and collaborate across compiler, runtime, and hardware teams to accelerate ML workloads on AWS accelerators.

You will optimize kernel performance across Neuron generations, perform profiling, and implement fusion, tiling, and scheduling while working with customers to tailor ML models for AWS accelerators.