Senior ML Engineer – LDM Behavioral Alignment for Driving

Rivian; Palo Alto (CA); On-site; USD 179,000 - 223,800; Full time

Benefits offered by this job

Annual performance bonus

Comprehensive health insurance

401(k) Plan

Job summary

Rivian is looking for a Senior Machine Learning Engineer in Palo Alto, California. This role focuses on driving behavioral optimization for the Large Driving Model (LDM). You'll be responsible for identifying performance bottlenecks, improving vehicle actions, and collaborating across teams to implement impactful solutions.

The ideal candidate has over 5 years of experience in ML, proficiency in Python, and experience with Reinforcement Learning. Rivian offers a competitive salary range of $179,000 to $223,800, along with comprehensive benefits.

Qualifications

• 5+ years of experience as an ML Engineer, Motion Planning Engineer, or Data Engineer.
• Advanced proficiency in Python and deep learning frameworks.
• Experience with Reinforcement Learning, reward design, and policy optimization.

Responsibilities

• Diagnose performance bottlenecks for driving behaviors and establish solution roadmaps.
• Design data mining strategies and metrics to inform model fine-tuning.
• Deliver RL algorithms and optimize on-road driving policies.

Skills

Python

Reinforcement Learning (RL)

Deep Learning frameworks (e.g., PyTorch)

Motion Planning

Data Engineering

Cross-Functional Communication

Education

B.S. or M.S. in Computer Science, Machine Learning, Robotics

Job description

Rivian is looking for a Senior Machine Learning Engineer in Palo Alto, California. This role focuses on driving behavioral optimization for the Large Driving Model (LDM). You'll be responsible for identifying performance bottlenecks, improving vehicle actions, and collaborating across teams to implement impactful solutions.

The ideal candidate has over 5 years of experience in ML, proficiency in Python, and experience with Reinforcement Learning. Rivian offers a competitive salary range of $179,000 to $223,800, along with comprehensive benefits.