Staff Machine Learning Engineer, Vehicle Behaviors

About the Team

The Vehicle Behaviors team is responsible for improving the decision-making capabilities and reliability of the AV in real world scenarios. Working broadly across domains and planning components, this team owns everything from ML-based Agent Predictions and Ego Trajectory Generation to safety-critical and heuristic-driven components like Trajectory Selection, to off-board metrics and tools used to measure performance and prevent regressions. This team has some of the most visible, high-impact work on the AV’s performance and plenty of challenging problems still to solve.

About the Role

We are looking for strong machine learning engineers to help in developing and deploying motion planning components for next generation self-driving systems. This requires strong ownership over critical components that cross domain boundaries. Candidates need to understand, experiment, improve, and field state-of-the-art machine learning systems in real-time, safety-critical applications. We are focused on building a product. Candidates should have a mission-driven mindset and customer-centric obsession to deliver a compelling product, and be able to work with significant cross-functional interactions. You might be a good fit for this role if you have experience with any of the following:

• Deploying state of the art ML models or architectures to solve planning problems on fielded products
• Refining ML outputs for use in down-stream planning components
• Integrating ML-based planning techniques side-by-side with heuristic approaches in production
• Building data mining, data introspection, or other critical ML tooling

What Success Looks Like

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
• Provide input in the technical direction for the team, and work cross-functionally to develop safe systems. This will include working closely with other teams such as perception, localization, and controls to ensure that the input to the motion planning modules is appropriate.
• 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
• Provide input to team roadmaps and ensure product features are properly prioritized.
• 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.

Experience:

• Strong Python skills
• Experience with key ML libraries like PyTorch and TensorRT
• Experience with cloud orchestration tools like Flyte
• Exceptional written and verbal communications skills
• Experience developing code for real-world robotics/semi-autonomous platform

#LI-AW1