Software Development Manager, ML Accelerators, AWS Neuron, Annapurna Labs

About the role

AWS Machine Learning accelerators are at the forefront of AWS innovation. The Trainium chip delivers industry-leading ML inference and training performance at the lowest cost in the cloud. This is enabled by edge software stack, the AWS Neuron Software Development Kit, which includes a Machine Learning compiler. The SDK natively integrates into popular ML frameworks, such as PyTorch, JAX and TensorFlow. AWS Neuron is widely adopted by many internal and external customers and partners.

Responsibilities

• Drive the design and implementation of ML compiler solutions
• Develop advanced optimization techniques
• Lead hardware bring-up for next-generation chips
• Influence pre-silicon design decisions
• Mentor your team

Requirements

• Bachelor's degree or foreign equivalent in Computer Science, Engineering, Mathematics, or a related field
• 5+ years of Software Engineer, Software Developer, or related occupational experience
• 3+ years of engineering team management experience
• Proven track record in new hardware bring-up
• Solid Knowledge of computer/chip architecture and code generation

Qualifications

• Experience with AI/ML technologies
• Experience in compiler design and architecture
• Knowledge of LLVM and MLIR

Skills

Our team puts a high value on work-life balance. It isn’t about how many hours you spend at home or at work; it’s about the flow you establish that brings energy to both parts of your life. We believe striking the right balance between your personal and professional life is critical to life-long happiness and fulfillment.

Benefits

• Comprehensive benefits including health insurance, 401(k) matching, paid time off, and parental leave
• Learn more about our benefits at https://amazon.jobs/en/benefits

Pay

The base salary range for this position is listed below. Your Amazon package will include sign-on payments and restricted stock units (RSUs). Final compensation will be determined based on factors including experience, qualifications, and location.

• USA, WA, Seattle - $184,900.00 - $250,200.00 USD annually

Schedule

Work/Life Balance Our team puts a high value on work-life balance. It isn’t about how many hours you spend at home or at work; it’s about the flow you establish that brings energy to both parts of your life. We believe striking the right balance between your personal and professional life is critical to life-long happiness and fulfillment. We offer flexibility in working hours and encourage you to find your own balance between your work and personal lives.