Staff Machine Learning Engineer, Motion Planning, Autonomy

$228,000 - $228,000/Yr

Rivian - Palo Alto, CA

Full-time - Senior

Palo Alto, CA

Transportation Equipment Manufacturing

About the position

The Staff Machine Learning Engineer for Motion Planning at Rivian will play a pivotal role in developing and deploying machine learning models that enhance the decision-making capabilities of the Rivian Autonomy stack. This position focuses on leveraging extensive driving data to improve the safety and experience of Rivian's customers, contributing significantly to the company's mission of creating emissions-free Electric Adventure Vehicles.

Responsibilities

• Examining all aspects of successful model design from data and supervision quality to model design and validation strategy.
• Taking charge of optimizing, supervising, and enhancing on-road performance for models moving towards production during the development phase.
• Recognized as a key technical authority within the department, offering guidance on current and future technology decisions.
• Possessing an in-depth grasp of the business and operational implications of various technology tradeoffs, effectively balancing them according to the company's needs.
• Empowering fellow engineers by crafting easily extendable and collaborative designs and code.
• Displaying the ability to influence and foster consensus, even during challenging technical discussions.

Requirements

• M.S. or Ph.D. in Machine Learning or a related field
• 6+ years of experience working on production ML systems
• Expert level knowledge and experience in Python and PyTorch or TensorFlow
• Solid understanding of state-of-the-art models and approaches in perception, prediction, or planning in the self-driving space
• Capable of working in a fast-paced development environment
• Good team player with excellent communication skills that can build team consensus
• Self-motivated with a demonstrated ability to set direction and guide the team towards it
• Hands-on approach with a proactive identification of gaps
• Good understanding of linear algebra, geometry, algorithms, and data structures
• Experience with Automotive Self-Driving or similarly complex Robotics Systems.

Benefits

• Robust medical/Rx, dental and vision insurance packages for full-time employees, their spouse or domestic partner, and children up to age 26. Coverage is effective on the first day of employment, and Rivian covers most of the premiums.