Senior Software Engineer, ML Training Infrastructure

Stack AV

$120K — $150K *
Transportation
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • Experience building end-to-end ML model pipelines including logs processing and feature extraction.
  • Proven track record of shipping ML products at scale, such as NLP or computer vision systems.
  • Ability to create abstractions and tools to facilitate rapid iteration on models for ML engineers.
  • Familiarity with autonomous vehicles and related technologies is a plus.

Responsibilities

  • Develop scalable, reliable ML infrastructure in a dynamic setting by collaborating with multiple modeling teams.
  • Build tools to enhance the developer experience for ML engineers, focusing on model testing and validation.
  • Optimize model performance and manage deployment processes to ensure reliability.
  • Understand and communicate design tradeoffs effectively within cross-functional teams.
  • Facilitate experimentation and improvements in model training and large data processing pipelines.

Benefits

  • Robust training framework to support professional growth in machine learning.
  • Collaborative work environment with a focus on innovation in autonomous vehicles.
  • Opportunities for hands-on experience with cutting-edge ML applications.
  • Support for experimentation and development in both model optimization and scaling.
Full Job Description
About the Role:

In the ML Training, our mission is to provide a reliable, scalable, and easy to use training framework for modeling needs of Stack AV. In addition, this team is responsible for the overall developer experience of ML engineers including building tools for testing, validation, and understanding models and the data used to train them. Finally, we are responsible for model optimization and deployment.

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.

Qualifications (Preferred):
  • 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

Similar Jobs

More Jobs at Stack AV

More Transportation Jobs

Find similar Senior Software Engineer, ML Training Infrastructure jobs: