The Principal Engineer role at Amazon focuses on driving AI model performance across the lifecycle on AWS Trainium, with a strong emphasis on pre-training, post-training, and inference serving. This position is part of the AWS Neuron SDK team within Annapurna Labs, which builds the software stack for Amazon's custom silicon. The role requires deep hands-on expertise in one of the key areas while influencing across all three. The team values technical leadership and collaboration to enhance model performance and ensure efficient deployment at scale.

• 10+ years of non-internship professional software development experience
• 7+ years with distributed training or RL systems at scale
• Strong proficiency in C/C++ and Python
• Experience with AI/ML model execution, inference serving, or compiler optimization
• Technical leadership and mentorship experience of engineering teams
• Deep hands-on expertise in ML model architecture, post-training methods, or AI inference optimization

• Drive AI model performance across the lifecycle on AWS Trainium.
• Optimize kernel performance for ML training and drive performance improvements.
• Deliver first-class reinforcement learning and preference optimization methods.
• Develop inference optimization strategies within the Neuron SDK.
• Serve as a technical leader on cross-functional projects spanning pre-training, post-training, and inference.
• Mentor senior engineers and influence technical direction across multiple teams.

• 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.