Staff ML Training Platform Architect

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

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.