Software Engineer, ML Platform (ML Training)

Build the Zoox Training framework leveraged by all ML teams within Zoox, ensuring it is highly scalable, reliable, and efficient. Design, implement, and operate a robust and efficient ML platform to enable the training, validation, serving, and monitoring of ML models. Collaborate closely with cross-functional teams, including ML researchers, software engineers, and data engineers, to define requirements and align on architectural decisions. This role supports foundational models and reinforcement learning across the ML Platform team, which owns the model repository and lifecycle management tools used by applied research teams for in- and off-vehicle use cases. You will significantly reduce the time from ideation to productionization of cutting-edge AI innovation as Zoox deploys its robotaxis on public roads.