Stack AV
Staff Software Engineer, ML Training Infrastructure
Pittsburgh, PA or Remote +1 · Remote · Staff · Full-time · Posted Oct 1, 2026
- Location
- Pittsburgh, PA or Remote
Pittsburgh/Remote - Experience
- 6+ years of experience with ML Platforms and building ML-based applications.
- Education
- Bachelor’s or Master’s degree in Computer Science, Engineering, or a related field.
- Visa sponsorship
- The position may be contingent on verifying a candidate’s residence, U.S. person status, and/or citizenship status. U.S. export control rules may require an export license, restrict employment, or prevent the candidate from working in the position.
About the job
Design and build a high-performance, multi-tenant ML training platform and end-to-end model pipelines to accelerate autonomous vehicle development.
What you'll do
- Design and evolve platform components for orchestration, training, observability, and performance.
- Deliver end-to-end ML pipelines spanning data processing, model training, profiling, and acceleration.
- Analyze infrastructure performance, improve tooling for ML engineers, and promote engineering excellence.
What they ask for
- Strong programming skills in Python.
- Experience with ML platforms and building ML-based applications.
- Experience with model training, model optimization, or large-scale data processing pipelines.
- Proven ability to build scalable, reliable infrastructure and collaborate across modeling teams.
Nice to have
- Prior experience with Lance, PyTorch, Ray or equivalent technologies.
- Autonomous vehicles (AV) experience; modeling experience is a bonus.
Tools
Python, Lance, PyTorch, Ray
AI focus
Build and operate infrastructure and pipelines for dataset curation, model training, validation, optimization, profiling, and deployment for autonomous vehicles.
Model-written summary · the original posting is authoritative