Staff Machine Learning Engineer, Software Vehicle Controls

$205,000 - $236,000/Yr

Rivian - Palo Alto, CA

Full-time - Senior

Palo Alto, CA

Transportation Equipment Manufacturing

About the position

The Staff Machine Learning Engineer at Rivian plays a pivotal role in developing software and data-driven solutions for electric vehicles. This position involves collaborating with engineers, data scientists, and domain experts to create machine learning models that enhance vehicle performance and optimize software parameters throughout the vehicle's lifecycle. The role emphasizes innovation in the automotive sector, leveraging big data from Rivian's customer and commercial fleets to push technological boundaries.

Responsibilities

• Design, implement, and optimize deep learning dataset generation, training, and evaluation tools and infrastructure.
• Develop domain-specific digital twins in collaboration with vehicle controls software engineers.
• Create statistical and machine learning models for estimating trip energy consumption, charging behavior, battery aging, and remaining useful life.
• Ship and deploy machine learning models for various vehicle domains to enhance efficiency across different operating conditions.
• Work cross-functionally with domain experts to deliver meaningful insights for electric vehicle controls and propulsion.
• Foster collaboration among engineering peers by creating adaptable designs and code.
• Facilitate consensus during complex technical discussions to maintain a productive working environment.
• Build deep learning models to enhance vehicle calibrations throughout its lifetime.

Requirements

• M.S. or Ph.D. in Machine Learning, Engineering, Computer Science, or related STEM disciplines.
• 5+ years of industry experience in machine learning systems.
• Expert-level proficiency in Python and Apache Spark.
• In-depth understanding of statistical and machine learning concepts, including deep and reinforcement learning.
• Strong knowledge in mathematical modeling, RNNs, CNNs, Transformers, LSTMs, reinforcement learning, imitation learning, GANs, and time-series analysis.
• Understanding of electric vehicle controls and propulsion, including high-voltage batteries and energy management.
• Excellent communication and interpersonal skills for both technical and non-technical audiences.
• Outstanding team player with the ability to build consensus in a dynamic environment.
• Proven ability to provide direction, guidance, and hands-on support to drive results.

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 effective on the first day of employment. Rivian covers most of the premiums.