Apply on Employer Site

Stack AV · 1 week ago

Staff Software Engineer, ML Training Infrastructure

United States

Full-time

Remote

Lead/Staff

Stack AV is developing revolutionary AI and advanced autonomous systems designed to enhance safety, reliability, and efficiency in the trucking transportation industry. The role involves providing a reliable, scalable training framework for modeling needs, improving developer experience for ML engineers, and handling model optimization and deployment.

Artificial Intelligence (AI)AutomotiveAutonomous VehiclesTransportation

No H1B

U.S. Citizen Only

Responsibilities

Experience with both ML Platforms and building ML-based applications

Experience building scalable, reliable infra at a fast-paced environment working with MLEs on several different modeling teams

A deep understanding of design tradeoffs and ability to articulate those tradeoffs and work with others on getting alignment

Experience with building ML models or ML infra in the domains of autonomous vehicles, perception, and decision making (desirable but not required)

Experience with model training, model optimization, or large data processing pipelines

Qualification

End to end ML model pipelineShipped ML productsModel optimizationBuilding ML-based applicationsLarge data processing pipelinesPrior AV experienceDesign tradeoffs understandingBuilding abstractionsTooling

Required

Built an end to end ML model pipeline including components such as logs processing, feature extraction, dataset storage, model configuration management, model training, experiment frameworks, and serving deployment

Shipped ML products (NLP, computer vision, recommender systems, etc.) at scale to make business impact

Knows how to build appropriate abstractions and tooling to ensure MLEs are able to rapidly iterate on models

Prior AV experience