The Senior Machine Learning Engineer role in the Distributed Training team for AWS Neuron focuses on the development, enablement, and performance tuning of various ML model families, particularly large-scale models like LLMs. The team collaborates closely with chip architects and engineers to optimize distributed training solutions on AWS custom silicon. Candidates should possess strong software development skills and a solid foundation in machine learning. The environment promotes mentorship and knowledge-sharing, aiming to support career growth.

• Bachelor's degree in computer science or equivalent.
• 5+ years of non-internship professional software development experience.
• 5+ years of programming experience with at least one software programming language.
• 5+ years of experience leading design or architecture of new and existing systems.
• 5+ years of experience in the full software development life cycle.
• Experience as a mentor or tech lead.
• Experience in machine learning and large-scale training with LLMs.
• Expertise in PyTorch.

• Lead efforts to build distributed training support into PyTorch and the Neuron compiler.
• Enable distributed training strategies to optimize models for peak performance.
• Maximize efficiency on AWS custom silicon, including Trainium servers.
• Collaborate effectively within cross-functional teams.
• Deep dive into technical challenges to deliver innovative solutions.

• Employees at Amazon are often offered comprehensive health benefits—including multiple medical plan options (no pre-existing condition exclusions, 100% covered in-network preventive care), dental and vision plans, a 24/7 medical advice line from day one, expert second-opinion services, and broad mental-health support with several free counseling sessions (including pediatric). Financial wellness typically includes a 401(k) with company match (up to 2%), Restricted Stock Units (equity), FSAs, an emergency savings program, product and partner discounts, and even college-savings and home-purchase programs. Overall, the package is designed to support employees and their families’ health, finances, and day-to-day life.