Senior Software Development Engineer, AWS Neuron Frameworks

Job Description

Develop and enhance support for PyTorch and JAX on AWS Neuron for machine learning accelerators.

Requirements

• 5+ years of non-internship professional software development experience
• Bachelor's degree in computer science or equivalent
• Experience developing Machine Learning infrastructure and/or ML Frameworks
• Experience with technologies and tools such as XLA, vLLM or Hugging Face transformers

Responsibilities

• Develop and extend support for leading ML frameworks on AWS Neuron
• Collaborate with AWS Neuron teams to optimize frameworks for accelerators
• Engage with PyTorch and JAX communities to improve performance and usability
• Set the direction for AWS Neuron framework technology roadmap

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)