Staff Machine Learning Engineer, Software Vehicle Controls

$205,000 - $236,000/Yr

Rivian - Irvine, CA

Full-time - Senior

Irvine, CA

Transportation Equipment Manufacturing

About the position

As a Staff Machine Learning Engineer at Rivian, you will play a pivotal role in developing advanced machine learning models to optimize vehicle performance and enhance the efficiency of electric vehicles. Collaborating with engineers, data scientists, and domain experts, you will leverage big data from our customer and commercial fleets to push the boundaries of technological innovation in the automotive sector.

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 well-structured 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.

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.