Sr. ML Kernel Performance Engineer, AWS Neuron, Annapurna Labs

Job Description

The Annapurna Labs team at Amazon Web Services (AWS) is seeking a highly skilled kernel engineer to optimize machine learning workloads for their custom ML accelerators.

Requirements

• 5+ years of non-internship professional software development experience
• Experience in leading design or architecture of new and existing systems
• Bachelor's degree in computer science or equivalent
• Experience working at the hardware-software boundary

Responsibilities

• Design and implement high-performance compute kernels for ML operations
• Analyze and optimize kernel-level performance across multiple generations of Neuron hardware
• Work directly with customers to enable and optimize their ML models on AWS accelerators
• Collaborate across teams to develop innovative kernel optimization techniques

Benefits

• Medical, dental, and vision coverage (multiple plan options)
• Health savings and flexible spending accounts
• 24/7 Employee Assistance Program for mental health support
• 401(k) with company match and various investment choices
• Company-paid life/AD&D and disability insurance
• Restricted stock units (RSUs) for employee ownership
• Child care and elder care referrals, adoption assistance
• Paid maternity and parental leave (including Ramp Back)
• Paid time off (PTO) and company holidays
• Employee discount on Amazon.com products
• Career Choice program (95% tuition and textbook coverage)