Senior Machine Learning Engineer, Vehicle Behaviors

Stack AV is developing revolutionary AI and advanced autonomous systems to enhance safety and efficiency in modern operations. They are seeking a Senior Machine Learning Engineer to develop and deploy motion planning components for next-generation self-driving systems, focusing on critical machine learning systems in real-time applications.

Responsibilities:

• Design, scope, implement, and integrate machine learning systems to solve on-vehicle behavior problems in a real-time, resource-constrained environment
• Develop tooling and infrastructure necessary to measure and iterate on ML models
• Work closely with systems engineers to ensure a safe, well tested product is delivered
• Work closely with verification teams to ensure proper testing and validation of the motion planning modules. Make extensive use of unit testing, simulation, and log simulation to properly validate their work
• Collaborate with other autonomy teams including perception, localization, controls, etc. to ensure solutions are appropriate for real-world performance
• Identify bottlenecks and limitations in system performance, and develop novel motion planning components to unlock new capabilities and ensure a reliable system
• Be involved in experimentation, design and iteration exercises, and help to align stakeholders by using strong presentation and communication skills

Requirements:

• Strong Python skills
• Experience with key ML libraries like PyTorch and TensorRT
• Experience with cloud orchestration tools like Flyte
• Strong written and verbal communications skills