Let’s get started
By clicking ‘Next’, I agree to the Terms of Service
and Privacy Policy, and consent to receive emails from Rise
Jobs / Job page
Senior/ Staff Machine Learning Engineer - Generative AI image - Rise Careers
Job details

Senior/ Staff Machine Learning Engineer - Generative AI

The Perception team at Zoox is at the forefront of leveraging GenAI to create synthetic data, unlocking scalable training and evaluation for our autonomous system's perception and entire stack. As a Generative AI Engineer, you will develop and train cutting-edge models for sensor-level scenario generation, utilizing world models and radiance fields techniques with large-scale proprietary data. This role directly impacts the productivity, safety, and capabilities of Zoox's autonomous system by validating algorithms in real-world conditions.


In this role, you will:
  • Define and execute the ML roadmap for synthetic data generation using generative AI, evolving both model and infrastructure to meet the training and evaluation needs of Zoox’s autonomous driving solution.
  • Lead the development of generative models from small scale objects to complete scenarios, from research all the way to deployment.
  • Design effective model architectures and sophisticated training techniques, leveraging all the inputs from our sensor stack and the overall large scale data we have at Zoox.
  • Collaborate with perception, planning, safety, simulation, and systems teams to integrate your models into our offline pipelines.
  • Validate and optimize your solutions using real-world driving scenarios, directly contributing to the safety and reliability of Zoox's autonomous system


Qualifications:
  • MS or PhD in Computer Science, Machine Learning, or related technical field
  • Demonstrated experience architecting, training and deploying large models such as diffusion, flow matching, GANs and/or NeRFs.
  • Experience building and maintaining ML training pipelines, including data preprocessing, model training, and evaluation
  • Proficiency in Python and ML libraries (PyTorch, NumPy) demonstrated through professional or research projects
  • Experience training with large scale datasets (e.g. tens of millions of videos)


Bonus Qualifications:
  • Publications in top-tier conferences (CVPR, ICCV, RSS, ICRA)
  • Experience with autonomous robotics systems
  • Experience implementing 4D Gaussian Splatting


$230,000 - $318,000 a year
Base Salary Range

There are three major components to compensation for this position: salary, Amazon Restricted Stock Units (RSUs), and Zoox Stock Appreciation Rights. A sign-on bonus may be offered as part of the compensation package. The listed range applies only to the base salary. Compensation will vary based on geographic location and level. Leveling, as well as positioning within a level, is determined by a range of factors, including, but not limited to, a candidate's relevant years of experience, domain knowledge, and interview performance. The salary range listed in this posting is representative of the range of levels Zoox is considering for this position.

Zoox also offers a comprehensive package of benefits, including paid time off (e.g. sick leave, vacation, bereavement), unpaid time off, Zoox Stock Appreciation Rights, Amazon RSUs, health insurance, long-term care insurance, long-term and short-term disability insurance, and life insurance.

Average salary estimate

$274000 / YEARLY (est.)
min
max
$230000K
$318000K

If an employer mentions a salary or salary range on their job, we display it as an "Employer Estimate". If a job has no salary data, Rise displays an estimate if available.

Similar Jobs
Photo of the Rise User
Abridge Hybrid No location specified
Posted 11 hours ago

Experienced backend engineer needed to develop scalable cloud-native applications powering generative AI healthcare solutions at Abridge, a forefront innovator in medical AI technology.

Photo of the Rise User
Posted 21 hours ago

Experienced Full Stack Engineer wanted to develop scalable, secure federal applications remotely with a strong Agile team using modern tech stacks like Java, React, and AWS.

Photo of the Rise User
Ingram Content Group Hybrid 1 Ingram Blvd, La Vergne, TN 37086, USA
Posted 23 hours ago

Experienced Senior Software Engineer role focusing on mainframe COBOL and DB2 development at Ingram Content Group in a hybrid work environment.

Photo of the Rise User
Posted 3 hours ago

Upside is looking for a Senior Software Engineer II to drive innovative, scalable solutions that enhance the user experience and business impact in a hybrid Austin-based role.

Photo of the Rise User

Lead the design and architecture of scalable backend fulfillment optimization systems at Coupang, a top global e-commerce company.

Photo of the Rise User
Madhive Hybrid No location specified
Posted 2 hours ago

Madhive is looking for a Frontend Engineer to build high-performance, user-friendly applications that connect local advertisers with communities at scale.

Photo of the Rise User

Lead the design and scaling of cloud-native cybersecurity services at Palo Alto Networks as a Principal Cloud Software Engineer.

Photo of the Rise User
Ridgeline Hybrid San Ramon, CA; Reno, NV; New York, NY
Posted 23 hours ago

Lead the creation of AI-driven frontend solutions as a Staff Frontend Engineer at Ridgeline, a pioneering investment management platform.

Posted 21 hours ago

Join Junior as a Senior Full Stack Engineer in NYC to build cutting-edge AI-driven investment research tools alongside a tight-knit growing team.

Photo of the Rise User
Voltage Park Hybrid No location specified
Posted 13 hours ago

Contribute to building scalable AI infrastructure as a Software Engineer at Voltage Park, focused on innovative solutions with NVIDIA GPU technology.

Posted 20 hours ago

Contribute as a Full Stack Engineer at Doe to build AI-powered platforms transforming operational workflows in a fast-paced, innovative environment.

Posted 22 hours ago

Experienced AI/ML Engineer sought by Cadmus to lead GenAI and traditional machine learning initiatives in scalable, cloud-based environments.

Photo of the Rise User
Ridgeline Hybrid San Ramon, CA; Reno, NV; New York, NY
Posted 22 hours ago

Senior Engineer role at Ridgeline to design and build AI-powered agents integrating cutting-edge generative AI with investment management platforms.

Zoox was founded to make personal transportation safer, cleaner, and more enjoyable—for everyone. To achieve that goal, the team created a whole new form of transportation. Zoox will provide mobility-as-a-service in dense urban environments.

3 jobs
MATCH
Calculating your matching score...
FUNDING
SENIORITY LEVEL REQUIREMENT
TEAM SIZE
EMPLOYMENT TYPE
Full-time, onsite
DATE POSTED
August 1, 2025
Risa star 🔮 Hi, I'm Risa! Your AI
Career Copilot
Want to see a list of jobs tailored to
you, just ask me below!