Member of Technical Staff - Machine Learning Engineer, AGI Autonomy

Amazon.com, Inc.•San Francisco, CA

33d

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

Nice To Haves

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

Stand Out From the Crowd

Upload your resume and get instant feedback on how well it matches this job.

Upload and Match Resume