This role involves leveraging engineering and machine learning expertise to develop generative AI for shopping at Amazon. The SFAI Training Infrastructure team focuses on building a unified platform for large-scale LLM training, addressing challenges in distributed systems and machine learning. The position offers the opportunity to work with talented scientists and engineers, contributing to innovative solutions that enhance customer experience. Candidates should be passionate about technology and ready to make a significant impact in the industry.

• 3+ years of non-internship professional software development experience
• 2+ years of non-internship design or architecture experience with new and existing systems
• Experience programming with at least one software programming language
• Knowledge of Machine Learning and LLM fundamentals, including transformer architecture and optimization techniques
• Knowledge of ML frameworks including JAX, PyTorch, and TensorRT
• Experience with CUDA/C++/Kernel development

• Design and implement a stable and efficient training system for model training and reinforcement learning.
• Collaborate with applied scientists and engineers to improve training efficiency and reliability.
• Design and implement scalable data infrastructure for Amazon-scale data ingestion and processing.
• Build end-to-end RL post-training pipelines at cluster scale.
• Improve RL training stability by monitoring and tuning key metrics.
• Optimize RL post-training efficiency through various techniques.
• Partner with research scientists to translate new RL algorithms into scalable systems.
• Profile and eliminate bottlenecks across compute, networking, and storage.
• Build observability systems for training dynamics and experiment tracking.
• Collaborate cross-functionally to run experiments and unblock research progress.

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