Stack AV
Staff Software Engineer, ML Training Infrastructure
Pittsburgh, PA or Remote · Staff+ · Contract
Stay score
odds of building a lasting career here
Thin sponsorship signal and lottery-bound. A low-probability bet with your clock running. Prioritize cap-exempt roles and proven entry-level sponsors first.
Lottery odds assume a STEM candidate.
Personalize to your clock →Employer immigration record
from this employer's Department of Labor filings
Files H-1B transfers
Sourced from Department of Labor LCA, PERM and prevailing-wage disclosure data. Employer matching is by name, so figures may be split across an employer's legal entities. Absence of a filing means none appears in our copy of the data, not that none exists.
Community outcomes
No reports yet — be the first to help the next applicant.
About the role
- The ML Training team is dedicated to increasing Stack's AV development velocity by accelerating machine learning iterations.
- We are responsible for the overall AI workflow, ranging from dataset curation to training, validation, acceleration, optimization and deployment of large-scale models that power our autonomous vehicles.
- In addition, this team is in charge of evangelizing best practices and frameworks among Machine Learning Engineers across the company.
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.
- 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
Requirements
- Bachelor's or Master's degree in Computer Science, Engineering, or a related field
- Prior experience with Lance, PyTorch, Ray or equivalent technologies
- Experience with model training, model optimization, or large-scale data processing pipelines
- Excellent verbal and written communication skills, with the ability to convey complex technical concepts to non-technical stakeholders
- 6+ years of experience with ML Platforms and building ML-based applications (modeling experience is a bonus).
- Strong programming skills in Python
- 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
- Autonomous vehicles (AV) experience is a bonus
- Strong analytical and problem-solving skills
Visa & Work Authorization
- As such, this position may be contingent upon Stack AV verifying a candidate's residence, U.S. person status, and/or citizenship status
This listing is sourced directly from Stack AV's careers page and normalized into a canonical job model.