Staff ML Platform Engineer — Training Infrastructure

Stack AV Co; United States; On-site; USD 170,000 - 230,000; Full time

Job summary

Stack AV Co is seeking a Staff ML Training engineer to drive the design and development of a high-performance, multi-tenant AI training platform across ML Platform, Infrastructure, Autonomy and Safety Evaluation teams. You will balance hands-on coding with long-term technical direction to accelerate AV development at Stack.

The role emphasizes end-to-end ML pipelines, observability and scalable workflows, with a focus on accelerating model iterations and evangelizing best practices across the

Qualifications

• Bachelor’s or Master’s degree in Computer Science, Engineering, or a related field.
• 6+ years of experience with ML Platforms and building ML-based applications (modeling experience is a bonus).
• Strong programming skills in Python, C++ or equivalent.
• Prior 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++

Strong communication

Analytical thinking

Problem solving

Cross-functional collaboration

English proficiency

Education

Bachelor’s or Master’s in Computer Science or related field

Tools

Lance

PyTorch

Ray Data

Job description

Stack AV Co is seeking a Staff ML Training engineer to drive the design and development of a high-performance, multi-tenant AI training platform across ML Platform, Infrastructure, Autonomy and Safety Evaluation teams. You will balance hands-on coding with long-term technical direction to accelerate AV development at Stack.

The role emphasizes end-to-end ML pipelines, observability and scalable workflows, with a focus on accelerating model iterations and evangelizing best practices across the