Staff Software Engineer, ML Training Infrastructure

Highlights

• Design and scale multi-tenant AI training platform for autonomous trucking.
• Own end-to-end ML pipelines from dataset curation to model deployment.
• Accelerate AV development by improving reliability and training velocity.

Ideal candidate

A collaborative infrastructure builder who cares about reliability, scalability, and developer velocity. Motivated by empowering MLEs and accelerating safe autonomous trucking.

Requirements

• Bachelor's or Master's degree in Computer Science, Engineering, or related field
• 6+ years of experience with ML Platforms and building ML-based applications
• Strong programming skills in Python
• Prior experience with Lance, PyTorch, Ray or equivalent technologies
• Proven track record building scalable, reliable infra with MLEs across teams
• Experience with model training, optimization, or large-scale data processing pipelines
• Strong analytical, problem-solving and communication skills