Member of Technical Staff ML Engineer, Frontier AI Robotics

We are seeking a highly skilled Machine Learning Systems Engineer to join Frontier AI Robotics team. This role focuses on building and optimizing distributed training infrastructure for large-scale machine learning models particularly in deep learning and transformer-based architectures. You will work closely with scientists and engineers to deliver scalable high-performance systems that power state-of-the-art AI research and applications.


About the team
At Frontier AI & Robotics were not just advancing robotics were reimagining it from the ground up. Our team is building the future of intelligent robotics through frontier foundation models and end-to-end learned systems. We tackle some of the most challenging problems in AI and robotics from developing sophisticated perception systems to creating adaptive manipulation strategies that work in complex real-world scenarios.

What sets us apart is our unique combination of ambitious research vision and practical impact. We leverage Amazons massive computational infrastructure and rich real-world datasets to train and deploy state-of-the-art foundation models. Our work spans the full spectrum of robotics intelligence from multimodal perception using images videos and sensor data to sophisticated manipulation strategies that can handle diverse real-world scenarios. Were building systems that dont just work in the lab but scale to meet the demands of Amazons global operations.

Join us if youre excited about pushing the boundaries of whats possible in robotics working with world-class researchers and seeing your innovations deployed at unprecedented scale.

- 3 years of non-internship professional software development experience
- 2 years of non-internship design or architecture (design patterns reliability and scaling) of new and existing systems experience
- Experience programming with at least one software programming language
- Design build and optimize machine learning infrastructure for large-scale training and inference.
- Apply PyTorch Python and C skills to engineer modular scalable ML systems.
- Evaluate and implement parallelism techniques such as data tensor model and pipeline parallelism.
- Monitor and optimize GPU memory and throughput for training large models efficiently.
- Collaborate cross-functionally with research data infra teams to integrate new models and features.
- Deep understanding of LLM algorithm and deep learning framework like PyTorch.
- Mathematics and Statistics: Strong understanding of linear algebra calculus probability and statistics.

- 3 years of full software development life cycle including coding standards code reviews source control management build processes testing and operations experience
- Bachelors degree in computer science or equivalent

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status disability or other legally protected status.

Los Angeles County applicants: Job duties for this position include: work safely and cooperatively with other employees supervisors and staff; adhere to standards of excellence despite stressful conditions; communicate effectively and respectfully with employees supervisors and staff to ensure exceptional customer service; and follow all federal state and local laws and Company policies. Criminal history may have a direct adverse and negative relationship with some of the material job duties of this position. These include the duties and responsibilities listed above as well as the abilities to adhere to company policies exercise sound judgment effectively manage stress and work safely and respectfully with others exhibit trustworthiness and professionalism and safeguard business operations and the Companys reputation. Pursuant to the Los Angeles County Fair Chance Ordinance we will consider for employment qualified applicants with arrest and conviction records.

Pursuant to the San Francisco Fair Chance Ordinance we will consider for employment qualified applicants with arrest and conviction records.

Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process including support for the interview or onboarding process please visit for more information. If the country/region youre applying in isnt listed please contact your Recruiting Partner.