Zoox Logo

Zoox

Senior Machine Learning Engineer - ML Agents and Planning

Reposted 5 Days Ago
Be an Early Applicant
Hybrid
Foster City, CA, USA
242K-290K Annually
Senior level
Hybrid
Foster City, CA, USA
242K-290K Annually
Senior level
The role involves developing deep learning models for autonomous driving using imitation and reinforcement learning, and collaborating on machine learning infrastructure and performance analysis.
The summary above was generated by AI
The Offline Driving Intelligence team is responsible for developing Foundation Models for ML Agents and planning, applying them off-vehicle to provide generalization capabilities to simulation and validation. Our team collaborates closely with the Planner, Simulation and Validation teams to develop and validate our driving performance. As an ML Agents and Planning Machine Learning Engineer you will work on the bleeding edge of the industry, developing novel machine learning pipelines and models to predict the behavior of other agents in the world and planning the best course of action for the ego vehicle.

In this role, you will...

  • You will develop new deep learning models that use imitation learning and reinforcement learning to generate driving plans for human-like agents.
  • You will work on novel techniques to estimate the quality of those driving plans along the dimensions of safety, progress, comfort and realism.
  • You will contribute to our large-scale machine learning infrastructure to discover new solutions and push the boundaries of the field
  • You will develop metrics and tools to analyze errors and understand improvements of our systems
  • You will collaborate with engineers on Perception, Planning,Simulation, and Validation to solve the overall Autonomous Driving problem.

Qualifications

  • PhD degree in computer science or related field +1y of professional experience (top tier publications can remove the need for the year of experience) or, MSc +5y of professional experience in a relevant field.
  • Experience in Planning and / or Prediction using Reinforcement Learning techniques 
  • Experience with training and deploying transformer-based model architectures
  • Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines
  • Fluency in Python with a basic understanding of C++

Bonus Qualifications

  • Top tier publications (NeurIPS, ICML, CVPR)

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.

HQ

Zoox Foster City, California, USA Office

4000 E 3rd Ave, Foster City, CA, United States, 94404

Zoox Foster City, California, USA Office

1149 Chess Drive, Foster City, CA, United States, 94404

Zoox Fremont, California, USA Office

47540 Kato Road, Fremont, CA, United States, 94538

Zoox San Francisco, California, USA Office

60 Broadway St, San Francisco, CA, United States, 94111

Similar Jobs

An Hour Ago
Hybrid
San Mateo, CA, USA
165K-185K Annually
Senior level
165K-185K Annually
Senior level
Fintech • Professional Services • Sales • Financial Services
Lead development, maintenance, and monitoring of credit risk models and loss forecasts. Extract and analyze large datasets with Python/SQL, automate reporting and dashboards, perform EDA and stress/sensitivity analyses, document audit-ready model deliverables, support model governance/validation, and communicate insights to stakeholders to inform credit policy and decisioning.
Top Skills: CklightboxGoogle Cloud PlatformOscilarPythonPython WidgetsSQLTableauTaktileXgboost
3 Hours Ago
Hybrid
37K-66K Hourly
Senior level
37K-66K Hourly
Senior level
Fintech • Financial Services
Grow and manage relationships with affluent customers by providing advisory, multi-product banking solutions across deposits, lending, investments, and home/business banking. Proactively acquire new customers, lead discovery-based planning, coordinate with Wealth/Home Lending/Business partners, support branch service needs, champion digital adoption, and maintain accurate documentation and regulatory compliance. Role requires obtaining and maintaining FINRA and state insurance licenses.
3 Hours Ago
Hybrid
San Mateo, CA, USA
Entry level
Entry level
Fintech • Financial Services
Please provide the full job description text (replace ${desc}) so I can extract requirements, salary, technologies, and other details accurately.

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account