Post your job offer for free on H1BConnect with no upfront cost!

Logo

Hire with Us
Rivian logo

Staff Software Engineer, ML Acceleration and Performance

Rivian

10/19/2025

Palo Alto, CA

Full-time

Salary: $179,000 - $285,000 a year


Job Description

Rivian is seeking a Staff Software Engineer for ML Acceleration and Performance to work on advanced machine learning algorithms for self-driving features of their Electric Adventure Vehicles.

Requirements

  • MS. or Ph.D. in Computer Science, Electrical, Mechanical, Aerospace Engineering or a related field
  • 7+ years of industrial experience in deep learning, heterogenous computing, ML accelerators, and compilers
  • Experience in model compression, neural architecture search, knowledge distillation, pruning, quantization, and deploying inference on embedded processors
  • Strong Python programming background and familiarity with PyTorch, TensorFlow, or MXNet
  • Experience with low-level subroutines like BLAS, CUDA Kernels, or C++ for implementing inference logic
  • Experience with hardware spec documents and performance profiling tools

Responsibilities

  • Develop, optimize, and deploy ultra-low latency deep learning/machine learning algorithms for Rivian ADAS and Autonomy use cases
  • Research state-of-the-art model compression and efficient model design techniques
  • Collaborate with low-level software and hardware architecture teams to optimize ML models for embedded platforms

Benefits

  • Employees at Rivian are often offered comprehensive health, dental, and vision insurance, a 401(k) with company match, ESPP and potential RSU grants, performance bonuses, and generous time off including paid parental and sick leave. Perks can include 24/7 mental-health coaching and therapy, Hinge Health for Anthem enrollees, tuition assistance and professional development, plus on-site food and a wide range of employee discounts on items like computers, mobile phones, home loans, and pet care.
Logo

© 2024 H1BConnect. All rights reserved.

Check out our sister site LatamDev for tech jobs in Latin America! 🌎