Machine Learning Engineer • Senior

Who you are

5+ years of experience in machine learning engineering, particularly in optimizing large-scale models for production environments — you've successfully transformed cutting-edge research into high-performance systems and understand the intricacies of deploying AI solutions at scale.

Deep expertise in CUDA and TensorRT — you have a proven track record of leveraging NVIDIA's acceleration stack to achieve significant performance improvements in model inference, and you are familiar with the challenges of optimizing complex architectures for real-world applications.

You thrive in collaborative environments, working closely with scientists and engineers to influence model architecture decisions — your ability to communicate complex technical concepts clearly ensures that you can effectively contribute to team discussions and drive projects forward.

Strong analytical skills and a data-driven mindset — you are adept at designing benchmarking frameworks to measure and optimize model performance, and you enjoy exploring innovative solutions to tackle ambitious performance targets.

What you'll do

Join Amazon's Frontier AI & Robotics team and play a pivotal role in driving inference optimization strategies for large-scale foundation models — your work will directly impact the efficiency and effectiveness of robotics applications at Amazon.

Collaborate closely with scientists to influence model architectures for optimal hardware utilization — your insights will help shape the future of robotics and AI at Amazon, ensuring that our models are designed with performance in mind from the ground up.

Design and implement efficient compilation pipelines for complex transformer architectures — you will be responsible for ensuring that our models can be deployed reliably and efficiently in production environments.

Develop comprehensive benchmarking frameworks to measure and optimize model performance — your analytical skills will be crucial in identifying areas for improvement and ensuring that our models meet the highest standards of performance.

Build robust monitoring solutions to ensure reliable model serving at scale — you will work on creating systems that can effectively monitor model performance and alert the team to any issues that arise, ensuring that our robotics applications run smoothly.

What we offer

At Amazon, you will be part of a team that is at the forefront of AI and robotics innovation — we offer competitive salaries and comprehensive benefits, including health insurance, retirement plans, and generous paid time off.

You will have the opportunity to work alongside world-renowned AI pioneers and contribute to groundbreaking projects that are shaping the future of technology — your work will have a direct impact on how robotics and AI are integrated into everyday life.

We encourage you to apply even if your experience doesn't match every requirement — we value diverse perspectives and are committed to fostering an inclusive environment where everyone can thrive.