Staff ML Training Platform Engineer

Stack AV Co; Pittsburgh (Allegheny County); Hybrid; USD 150,000 - 190,000; Full time

Job summary

Stack AV Co in Pittsburgh is seeking a Staff Software Engineer to design and build a high‑performance, multi‑tenant ML training platform. You will balance hands‑on coding with technical leadership across ML Platform, Infrastructure, Autonomy and Safety Evaluation teams.

You will own end‑to‑end training pipelines, optimize performance, promote engineering excellence, and collaborate with MLEs to accelerate model iterations.

Qualifications

• 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 in Python and C++ or equivalent.
• Experience with Lance, PyTorch, Ray Data or equivalent technologies.
• Proven track record of building scalable, reliable infra in a fast-paced environment while working with MLEs across multiple modeling teams.
• A deep understanding of design tradeoffs and ability to articulate those tradeoffs to build alignment across XFN teams.
• Experience with model training, model optimization, or large-scale data processing pipelines.
• Autonomous vehicles (AV) experience is a bonus.
• Strong analytical and problem-solving skills.
• Excellent verbal and written communication skills, with the ability to convey complex technical concepts to non-technical stakeholders.

Responsibilities

• Design & evolve high‑performance training platform components including orchestration, training abstractions, control plane, observability and performance tuning.
• Deliver end‑to‑end ML model pipelines across logs processing, feature extraction, dataset schema design/storage, model configuration management, model training, and profiling/acceleration workflows.
• Analyze training infrastructure performance to identify and resolve performance bottlenecks
• Evangelize system abstractions and tooling that enable MLEs to rapidly iterate on models.
• Incorporate OSS tools to enable ML engineers self‑sufficiently profile and optimize their workflows
• Promote Engineering Excellence: Maintain a high bar for engineering excellence in their own work but also set a culture of engineering excellence within the team.

Skills

Python

C++

ML Platforms

Lance

PyTorch

Ray Data

System design

Analytical thinking

Communication

Cross-functional collaboration

Education

Bachelor’s or Master’s degree in Computer Science, Engineering, or related field

Tools

N/A

Job description

Stack AV Co in Pittsburgh is seeking a Staff Software Engineer to design and build a high‑performance, multi‑tenant ML training platform. You will balance hands‑on coding with technical leadership across ML Platform, Infrastructure, Autonomy and Safety Evaluation teams.

You will own end‑to‑end training pipelines, optimize performance, promote engineering excellence, and collaborate with MLEs to accelerate model iterations.