Senior/Staff Software Engineer - Machine Learning Infrastructure

Foster City, CA

Full Time Senior-level / Expert
Zoox logo
Apply now Apply later

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
  • 10-20+ 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
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.
Follow us on LinkedIn
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.
Job region(s): North America
Job stats:  5  1  0
  • Share this job via
  • or

Explore more AI/ML/Data Science career opportunities