Staff AI/ML Engineer, Vehicle Controls
Rivian
About Us Rivian and Volkswagen Group Technologies is a joint venture between two industry leaders with a clear vision for automotive’s next chapter. From operating systems to zonal controllers to cloud and connectivity solutions, we’re addressing the challenges of electric vehicles through technology that will set the standards for software-defined vehicles around the world. The road to the future is uncharted. By combining our expertise across connectivity, AI, security and more, we’ll map a new way forward. Working together, we’ll create a future that’s more connected, more intelligent, more sustainable for everyone. Role Summary At Rivian, our mission revolves around developing vehicles powered by software and data. As a Staff Machine Learning Engineer, you will be a crucial part of our team, working alongside engineers, data scientists, and domain experts to build models at scale, predict vehicle performance, and optimize SW parameters throughout the vehicle's lifetime. We leverage the power of big data collected from our extensive customer and commercial fleets to achieve this. We are looking for a Staff Machine Learning Engineer specializing in Deep Learning to build cutting-edge models for vehicle intelligence, anomaly detection, and predictive maintenance. If you thrive in a research-driven, applied AI environment, this is the perfect opportunity to push the boundaries of GANs, VAEs, Transformers, and Autoencoders in real-world applications. Responsibilities Develop and optimize RL-based models for vehicle control, smart scheduling, and dynamic decision-making. Apply Imitation Learning to enhance vehicle behavior prediction and self-adaptive AI systems. Architect Generative AI models (GANs, VAEs, Transformers, Autoencoders) for anomaly detection, predictive maintenance, and real-time vehicle intelligence. Build self-learning AI systems that improve vehicle performance over time through continuous learning from fleet data. Design unsupervised & self-supervised learning models to process high-dimensional vehicle telemetry data. Collaborate with data scientists, software engineers, and vehicle system experts to deploy AI-powered automotive solutions. Work on LLM-based automation tools, including AI-driven requirement management and chatbot-based vehicle control support. Improve MLOps pipelines for efficient AI/ML deployment and lifecycle management. Qualifications M.S. or Ph.D. in Machine Learning, AI, Computer Science, or a related STEM discipline. Expertise in Reinforcement Learning (RL, RLHF, RLAIF) and Imitation Learning. Experience developing Generative AI models (GANs, VAEs, Transformers, Autoencoders). Proficiency in deep learning frameworks (TensorFlow, PyTorch). Strong software engineering skills in Python, Apache Spark, and cloud-based ML frameworks (AWS/GCP/Azure). Experience optimizing AI models for real-time decision-making and large-scale deployment. Pay Disclosure Salary Range/Hourly Rate for Palo Alto, California Based Applicants: $146,900 - 183,600 USD (actual compensation will be determined based on experience, location, and other factors permitted by law). Salary Range/Hourly Rate for California Based Applicants: $135,100 - 183,600 USD (actual compensation will be determined based on experience, location, and other factors permitted by law). Benefits Summary: Rivian and Volkswagen Group Technologies provides robust medical/Rx, dental and vision insurance packages for full-time employees, their spouse or domestic partner, and children up to age 26. Coverage is effective on the first day of employment. Equal Opportunity Rivian and Volkswagen Group Technologies is committed to creating a diverse environment and is proud to be an equal opportunity employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, national origin, ancestry, sex, sexual orientation, gender, gender expression, gender identity, genetic information or characteristics, physical or mental disability, marital/domestic partner status, age, military/veteran status, medical condition, or any other characteristic protected by law. We are also committed to ensuring compliance with all applicable fair employment practice laws regarding citizenship and immigration status. Rivian and Volkswagen Group Technologies is committed to ensuring that our hiring process is accessible for persons with disabilities. If you have a disability or limitation, such as those covered by the Americans with Disabilities Act, that requires accommodations to assist you in the search and application process, please email us at candidateaccommodations@rivian.com. Candidate Data Privacy Rivian and VW Group Technologies (“Rivian and Volkswagen Group Technologies”) may collect, use and disclose your personal information or personal data (within the meaning of the applicable data protection laws) when you apply for employment and/or participate in our recruitment processes (“Candidate Personal Data”). This data includes contact, demographic, communications, educational, professional, employment, social media/website, network/device, recruiting system usage/interaction, security and preference information. Rivian and Volkswagen Group Technologies may use your Candidate Personal Data for the purposes of (i) tracking interactions with our recruiting system; (ii) carrying out, analyzing and improving our application and recruitment process, including assessing you and your application and conducting employment, background and reference checks; (iii) establishing an employment relationship or entering into an employment contract with you; (iv) complying with our legal, regulatory and corporate governance obligations; (v) recordkeeping; (vi) ensuring network and information security and preventing fraud; and (vii) as otherwise required or permitted by applicable law. Rivian and Volkswagen Group Technologies may share your Candidate Personal Data with (i) internal personnel who have a need to know such information in order to perform their duties, including individuals on our People Team, Finance, Legal, and the team(s) with the position(s) for which you are applying; (ii) Rivian and Volkswagen Group Technologies affiliates; and (iii) Rivian and Volkswagen Group Technologies’ service providers, including providers of background checks, staffing services, and cloud services. Rivian and Volkswagen Group Technologies may transfer or store internationally your Candidate Personal Data, including to or in the United States, Canada, and the European Union and in the cloud, and this data may be subject to the laws and accessible to the courts, law enforcement and national security authorities of such jurisdictions. Please note that we are currently not accepting applications from third party application services.
Confirm your E-mail: Send Email
All Jobs from Rivian