Staff Software Engineer, ML Training Infrastructure (Fully Remote)

Stack AV Co; United States; Remote; USD 120,000 - 180,000; Full time

Job summary

Stack AV offers a pivotal role shaping the training system for autonomous vehicle ML, delivering reliable and scalable solutions. You’ll design high-performance components, and drive end-to-end ML pipelines from data curation through training to deployment.

Join a team that leads across ML Platform, Infrastructure, Autonomy, and Safety, fostering innovation and inclusion while tackling real-world safety challenges in autonomous trucking.

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.

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.
• Deliver a reliable, scalable, user-friendly training system for autonomous vehicle ML.
• Drive technical direction across ML Platform, Infrastructure, Autonomy, and Safety teams.

Skills

Python

C++

PyTorch

Ray Data

Lance

ML Platforms

Education

Bachelor’s or Master’s in CS

Job description

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