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
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.