Senior Machine Learning Engineer, Auto Labeling

Summary

Join Stack's Auto Labeling team and contribute to the development of large ML models for high-quality labeled data generation. You will design, evaluate, and deploy causal large ML models using multi-sensor inputs, working with state-of-the-art techniques in large-scale distributed model training and evaluation. Collaborate with onboard and offboard downstream consumers throughout the development lifecycle. Contribute to building safety-critical and robust onboard and offboard perception systems. This role requires strong Python engineering skills, particularly PyTorch, and experience with distributed training systems. A degree in a related field and 3+ years of experience building ML systems are required.

Requirements

• Degree in Computer Science, Machine Learning, Robotics, or related field
• 3+ years building ML systems
• Experience with distributed training systems
• Strong Python engineering skills, particularly PyTorch
• Ability to work in a fast-paced startup environment while maintaining high-quality standards

Responsibilities

• Design, evaluate, and deploy a causal large ML models for high-quality labeled data generation from multi-sensor inputs
• Work with state-of-the-art techniques in large-scale distributed model training and evaluation
• Collaborate with onboard and offboard downstream consumers throughout the development lifecycle, from requirements gathering to evaluation and ongoing communication
• Contribute to the overarching vision of building safety-critical and robust onboard and offboard perception systems
• Uphold Stack’s culture of engineering excellence

Preferred Qualifications

• Master's or PhD preferred
• Autonomous vehicle space experience is a plus
• Strong experience with GPUs; CUDA experience is a plus