Zoox’s Machine Learning Infrastructure team operates in 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 improve ML models in collaboration with the applied research teams. We advocate for strong software engineering fundamentals with our customer teams.  We catalogue the features, quirks, limitations, and bugs present in the systems we maintain so that we can best advise our customer teams when they encounter them.

ML Infra is looking for focused, proactive senior-to-staff software engineers.  By joining our team, you will design and deliver a performant, comprehensive platform for ML research and development to our internal customers. You will improve the production deployment mechanisms used for on- and off-vehicle inference. You will work with teams from IT, build and release, and cluster management on projects to advance our mission. You will provide technical leadership to the team and mentor new hires. People management opportunities will be available as the team grows.


  • Integrate ML tools and frameworks into the Zoox software ecosystem
  • Design and implement in-house tools, utilities, and automation to improve ML research workflows
  • Advise and assist Zoox’s ML research teams with development on the ML infra platform
  • Lead internal and cross-team projects for ML infra


  • Experience with Python and Linux
  • 5+ years of professional software engineering experience
  • Strong software engineering fundamentals
  • Aptitude for troubleshooting, debugging, and fixing arcane issues

Bonus Qualifications

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

Vaccine Mandate

Employees working in this position will be required to be fully vaccinated against the COVID-19 virus. An applicant is considered fully vaccinated two weeks after their second dose in a 2-dose series, such as the Pfizer or Moderna vaccines, or two weeks after a single-dose vaccine, such as Johnson & Johnson’s Janssen vaccine. Applicants 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.