Staff Software Engineer, ML Platform

Stack develops revolutionary AI and autonomous systems for enhancing safety, reliability, and efficiency in trucking transportation industry.
Machine Learning
Staff Software Engineer
Remote
101 - 500 Employees
8+ years of experience
AI · Automotive

Description For Staff Software Engineer, ML Platform

Stack AV is at the forefront of developing revolutionary AI and autonomous systems for the trucking transportation industry. As a Staff Software Engineer on the ML Training team, you'll be instrumental in building and maintaining the company's machine learning infrastructure. The role combines deep technical expertise in ML platforms with hands-on development of scalable solutions. You'll be responsible for creating reliable training frameworks, improving developer experience, and optimizing model deployment processes. The position offers the flexibility of working remotely or from Pittsburgh, PA, and puts you at the intersection of autonomous vehicles and cutting-edge ML technology. This is an opportunity to shape the future of autonomous trucking while working with state-of-the-art technologies like CUDA, Python, and C++. The role requires a unique blend of ML platform expertise and practical engineering skills, making it perfect for someone who wants to impact the future of transportation technology.

Last updated 4 hours ago

Responsibilities For Staff Software Engineer, ML Platform

  • Provide reliable, scalable, and easy to use training framework for modeling needs
  • Build tools for testing, validation, and understanding models and training data
  • Handle model optimization and deployment
  • Work on model training and large data processing pipelines

Requirements For Staff Software Engineer, ML Platform

Python
  • Experience with both ML Platforms and building ML-based applications
  • Experience building scalable, reliable infrastructure in a fast-paced environment
  • Ability to work across teams
  • Experience building or using ML infrastructure for multiple customer teams
  • Deep understanding of design tradeoffs and ability to articulate them
  • Experience with model training, model optimization, or large data processing pipelines

Interested in this job?

Jobs Related To Stack AV Staff Software Engineer, ML Platform

Engineer, Staff AI/Machine Learning

Staff AI/Machine Learning Engineer position at Qualcomm developing embedded AI solutions for automotive platforms, requiring expertise in C/C++, ML frameworks, and embedded systems.

Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Senior Compiler Engineer role at AWS's Annapurna Labs, focusing on ML compiler development for custom AI accelerators with competitive compensation and hybrid work model.

Senior Staff Software Engineer

Senior Staff Software Engineer position at Intuit focusing on AI-native applications and distributed systems development with competitive compensation.

Computer Vision Systems Engineer, up to Staff

Computer Vision Systems Engineer role at Qualcomm developing advanced vision algorithms and architectures for Snapdragon platforms across mobile, automotive, and AR/VR applications.

Machine Learning Engineer - Conversation AI

Machine Learning Engineer role at DoorDash focusing on Conversation AI, LLM systems, and personalization for support chatbot development.