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