SDE- ML Engineer, Frontier AI Robotics

Amazon
San Francisco, California, USA2025-11-17ONSITE

About the job

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.

Responsibilities

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.

Qualifications

Minimum

3+ years of non-internship professional software development experience

2+ years of non-intternship design or architecture (design patterns, reliability and scaling) of new and existing systems experience

Experience programming with at least one software programming language

Preferred

3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience

Bachelor's degree in computer science or equivalent