Zoox’s Machine Learning Infrastructure team sits at the intersection of build engineering, internal software development, and applied ML research. We maintain the base layer of ML tools and deep learning frameworks used by the research teams, add additional features, tooling, automation, and applications, and do ML model improvement projects in collaboration with the applied research teams. We advocate for strong software engineering fundamentals, and aim to have comprehensive knowledge of the features, quirks, limitations, and bugs present in the systems we maintain.

ML Infra is looking for focused, proactive software engineers to join our team and help deliver a performant, comprehensive platform for ML research and development to our internal customers.


  • Integrate ML tools and frameworks into the Zoox software ecosystem
  • Design and implement in-house tools, utilities, and automation to improve ML research workflows
  • Provide expert advice and development assistance to Zoox’s ML research teams


  • 3+ years of industry experience
  • Experience with Python and Linux
  • Bachelor's degree in computer science (or related field) or equivalent
  • Strong software engineering fundamentals

Bonus Qualifications

  • Experience with C++ or CUDA, gdb, valgrind and related tools
  • Familiarity with TensorFlow or PyTorch
  • Hands-on experience with machine learning model development or deployment
  • Familiarity with TensorRT, Bazel, AWS EC2
*Hiring at multiple levels!

Vaccine Mandate

Employees working in this position will be required to have received a single dose of the J&J/Janssen COVID-19 vaccine OR have completed the two-dose Pfizer or Moderna vaccine series. In addition, employees will be required to receive a COVID-19 booster vaccine by the earlier of (1) April 15 or (2) within two months of becoming eligible for the booster vaccine.

Employees will be required to show proof of vaccination status upon receipt of a conditional offer of employment. That offer of employment will be conditioned upon, among other things, an Applicant’s ability to show proof of vaccination status. Please note the Company provides reasonable accommodations in accordance with applicable state, federal, and local laws.

About Zoox

Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market. Sitting at the intersection of robotics, machine learning, and design, Zoox aims to provide the next generation of mobility-as-a-service in urban environments. We’re looking for top talent that shares our passion and wants to be part of a fast-moving and highly execution-oriented team.

A Final Note:
You do not need to match every listed expectation to apply for this position. Here at Zoox, we know that diverse perspectives foster the innovation we need to be successful, and we are committed to building a team that encompasses a variety of backgrounds, experiences, and skills.