Mid-Level Machine Learning Engineer - Generative AI
Jobright.ai
Foster city,
California,
United States
Information Technology
Engineering
H-1B
Job posted on August 2, 2025
APPLY NOW
Job Description:
Verified Job On Employer Career Site
Job Summary: Zoox is at the forefront of leveraging Generative AI to enhance its autonomous systems. As a Generative AI Engineer, you will develop and train models for synthetic data generation, directly impacting the productivity and safety of Zoox's technology.
Responsibilities: • 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:
Required: • 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)
Preferred: • Publications in top-tier conferences (CVPR, ICCV, RSS, ICRA) • Experience with autonomous robotics systems • Experience implementing 4D Gaussian Splatting
Company: Zoox is an AI robotics company that provides mobility-as-a-service and self-driving car services. It is a sub-organization of Amazon. Founded in 2014, the company is headquartered in Foster City, California, USA, with a team of 1001-5000 employees. The company is currently Late Stage. Zoox has a track record of offering H1B sponsorships.
APPLY NOW
Job Summary: Zoox is at the forefront of leveraging Generative AI to enhance its autonomous systems. As a Generative AI Engineer, you will develop and train models for synthetic data generation, directly impacting the productivity and safety of Zoox's technology.
Responsibilities: • 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:
Required: • 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)
Preferred: • Publications in top-tier conferences (CVPR, ICCV, RSS, ICRA) • Experience with autonomous robotics systems • Experience implementing 4D Gaussian Splatting
Company: Zoox is an AI robotics company that provides mobility-as-a-service and self-driving car services. It is a sub-organization of Amazon. Founded in 2014, the company is headquartered in Foster City, California, USA, with a team of 1001-5000 employees. The company is currently Late Stage. Zoox has a track record of offering H1B sponsorships.