Machine Learning Engineer, Frontier AI & Robotics

Join the next revolution in robotics at Amazon's Frontier AI & Robotics team, where you'll contribute to breakthrough foundation models run at production scale. As a Software Development Engineer embedded in our science team, you'll be instrumental in transforming novel research into high-performance production systems.

You'll collaborate directly with scientists to optimize large-scale transformer architectures for robotics applications, leveraging your expertise in CUDA and TensorRT to achieve unprecedented inference efficiency at Amazon scale. In this role, you'll balance deep technical optimization work with strategic input on model architecture decisions, ensuring our innovative robotics models are designed with performance in mind from the ground up.

Key Responsibilities:

• Drive inference optimization strategies for large-scale foundation models using TensorRT, CUDA, and other NVIDIA tools
• Collaborate with scientists to influence model architectures for optimal hardware utilization
• Design and implement efficient compilation pipelines for complex transformer architectures
• Develop comprehensive benchmarking frameworks
• Build robust monitoring solutions for reliable model serving at scale

The Frontier AI & Robotics team is building the future of intelligent robotics through frontier foundation models and end-to-end learned systems. We tackle challenging problems in AI and robotics, from developing sophisticated perception systems to creating adaptive manipulation strategies. We leverage Amazon's massive computational infrastructure and rich real-world datasets to train and deploy state-of-the-art foundation models.

This is an opportunity to work at the intersection of cutting-edge AI research and practical robotics applications, with the chance to see your innovations deployed at unprecedented scale. The role offers competitive compensation ranging from $129,300 to $223,600 per year based on location, plus comprehensive benefits including equity and sign-on payments.