Staff Software Engineer, ML Training Infrastructure (Fully Remote)

• Deliver a reliable, scalable, user-friendly training system for autonomous vehicle ML.
• Drive technical direction across ML Platform, Infrastructure, Autonomy, and Safety teams.

Responsibilities:

• Design high-performance training platform components including orchestration and control plane.
• Deliver end-to-end ML pipelines from dataset curation to training and deployment.

Qualifications:

• Bachelor’s or Master’s in CS with 6+ years in ML platforms.
• Strong Python/C++ and experience with PyTorch, Ray Data, or Lance.

Stack AV

Stack is developing revolutionary AI and advanced autonomous systems for safer, more reliable, and efficient operations, focusing on autonomous trucking. The Stack team brings decades of experience in real-world systems and is committed to a culture of inclusion, entrepreneurship, and innovation.