Software Engineer - Machine Learning Platform



Software Engineering
Foster City, CA, USA
Posted on Thursday, July 8, 2021
Zoox’s Machine Learning Platform 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 Platform 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 platform
  • Lead internal and cross-team projects for ML platform


  • 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
There are three major components to compensation for this position: salary, Amazon Restricted Stock Units (RSUs), and Zoox Stock Appreciation Rights. The salary range for this position is $196,000 to $319,000. A sign-on bonus may be offered as part of the compensation package. 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.
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.