Staff ML Engineer, Hardware Software Co-Design
Rivian · Palo Alto, CA · Mechanical & Electrical Engineering
About this role
Rivian is hiring a staff-level Machine Learning Engineer based in Palo Alto, CA. The posting calls out experience with Python, CUDA, TensorFlow, PyTorch. Listed education preference: a master's degree or equivalent. Compensation is listed at $228,000–$285,000 per year.
- Role
- Machine Learning Engineer
- Function
- machine learning
- Level
- staff
- Track
- Tech leadership
- Location
- Palo Alto, CA
- Education
- Master's degree
- Department
- Mechanical & Electrical Engineering
- Posted
- Mar 18, 2026
More roles at Rivian
Job description
from Rivian careersAbout Rivian Rivian is on a mission to keep the world adventurous forever. This goes for the emissions-free Electric Adventure Vehicles we build, and the curious, courageous souls we seek to attract. As a company, we constantly challenge what’s possible, never simply accepting what has always been done. We reframe old problems, seek new solutions and operate comfortably in areas that are unknown. Our backgrounds are diverse, but our team shares a love of the outdoors and a desire to protect it for future generations. Role Summary We are looking for an Engineer / Research Scientist with deep expertise in quantized deep learning models for hardware acceleration in autonomous systems. In this cross-disciplinary role, you will bridge perception model design and hardware-aware deployment, enabling efficient execution of high-performance perception algorithms across embedded compute platforms. You will focus on researching state of the art perception models and develop optimization pipelines for the quantized versions of these models customized to provide real-time performance and energy efficiency on next-generation autonomy hardware. Responsibilities Research state of the art perception models in collaboration with the ADAS SW teams Lead the development of optimizations for mapping quantized perception models (e.g., CNNs, Transformers, LLMs) to embedded and…