Senior/Staff Software Engineer, ML Performance Optimization

Zoox is on a mission to reimagine transportation by building autonomous robotaxis that are safe and reliable. The ML Performance Optimization team is seeking a Senior/Staff Software Engineer to lead initiatives that enhance the efficiency of their Training and Inference platform, collaborating with various teams to drive ML innovation in autonomous driving.

Responsibilities:

• Develop and execute a strategic vision for the ML Performance Optimization team to unlock ML innovation in autonomous driving and rider experience
• Lead the design, implementation, and operation of cutting-edge ML Training and inference performance optimization techniques
• Collaborate closely with x-functional teams, including ML researchers, software engineers, data engineers, and hardware engineers, to define requirements and align on architectural decisions
• Enable the engineers in the team to grow their careers by providing technical guidance and mentorship