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