The Senior Software Development Engineer will join the Data & Intelligence Foundation team at Amazon, focusing on machine learning operations. This role involves designing and building ML training infrastructure that supports robot learning at scale. The team is dedicated to developing the foundational ML platform for autonomous robotics within Amazon's fulfillment network. Engineers will work closely with research scientists to operationalize advanced ML models and ensure efficient training workflows.

• 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
• Experience as a mentor, tech lead, or leading an engineering team
• 5+ years of experience in the full software development life cycle
• Bachelor's degree in computer science or equivalent
• Knowledge of Machine Learning and LLM fundamentals
• Knowledge of ML frameworks including JAX, PyTorch, and TensorRT

• Design and implement scalable ML training infrastructure on Kubernetes with GPU scheduling and fault-tolerant distributed training
• Build and maintain CI/CD pipelines for ML models from data ingestion through deployment
• Develop tooling for experiment tracking, hyperparameter optimization, and reproducibility
• Architect data pipelines that handle large-scale robotics datasets
• Collaborate with research scientists to operationalize novel ML models into production
• Establish monitoring, alerting, and observability for training workloads and model performance
• Drive best practices for GPU fleet management, cost optimization, and capacity planning

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