Member of Technical Staff - Machine Learning Engineer, AGI Autonomy

Amazon • San Francisco, CA

4d

About The Position

The AGI Autonomy Perception team performs applied machine learning research, including model training, dataset design, pre- and post- training. We train Nova Act, our state-of-the art computer use agent, to understand arbitrary human interfaces in the digital world. We are seeking a Machine Learning Engineer who combines strong ML expertise with software engineering excellence to scale and optimize our ML workflows. You will be a key member on our research team, helping accelerate the development of our leading computer-use agent. We are seeking a strong engineer who has a passion for scaling ML models and datasets, designing new ML frameworks, improving engineering practices, and accelerating the velocity of AI development. You will be hired as a Member of Technical Staff.

Requirements

• 3+ years experience building and deploying machine learning systems
• Experience with ML frameworks (e.g. PyTorch, TensorFlow) and ML orchestration tools
• Proven track record of improving ML systems, performance, and workflows
• Experience building and optimizing production-grade AI systems, e.g., for self-driving, simulation, or LLM applications
• Willingness to step outside typical role boundaries to get things done — every member of technical staff is expected to write code, design experiments, and interpret results
• Capacity to work autonomously and with a small team to drive research progress

Responsibilities

• Design, build, and deploy machine learning models, frameworks, and data pipelines
• Optimize ML training, inference, and evaluation workflows for reliability and performance
• Evaluate and improve ML model performance and metrics
• Develop tools and infrastructure to enhance ML development productivity

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Education Level: No Education Listed
• Number of Employees: 5,001-10,000 employees