Find a job 
Featured Companies 
Featured Universities 
For Companies At a Glance
- Tasks: Design and deploy cutting-edge deep learning models for our electric vehicles.
- Company: Join Rivian, a trailblazer in emissions-free adventure vehicles, committed to innovation and sustainability.
- Benefits: Enjoy flexible work options, competitive pay, and a culture that values the outdoors.
- Why this job: Be part of a mission-driven team that challenges the status quo and makes a real impact.
- Qualifications: PhD in relevant fields with expertise in PyTorch and model training frameworks required.
- Other info: Rivian is an equal opportunity employer, ensuring accessibility for all applicants.
The predicted salary is between 43200 - 72000 £ per year.
About Us
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.
Responsibilities
- Design, train, and deploy large deep learning models that can leverage the vast amount of labeled and unlabeled data from a fleet of million vehicles.
- Improve ecosystem for training infrastructure and deployment pipeline to accelerate model iteration and improve performance.
Qualifications
- PhD in CS/CE/EE, or equivalent, in industry experience.
- Deep knowledge of PyTorch.
- Experience with Cuda or Triton language for writing custom ops.
- Knowledge of model training framework (e.g. PyTorch Lightning).
- In-depth knowledge of transformer architecture and ways to accelerate the training and inference of transformer models.
- Experience of performing large scale distributed training of models.
- A track record of profiling model and doing detective work to improve model training and inference speed.
Preferred Qualifications:
- Previous experience in the autonomous driving industry.
- Knowledge of Nvidia TensorRT.
- Experience with edge computing systems.
- Knowledge of model optimization including quantization, pruning, etc.
Company Statements
Equal Opportunity
Rivian is an equal opportunity employer and complies with all applicable federal, state, and local fair employment practices laws. 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. Rivian 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.
Senior Machine Learning Engineer, AI Infrastructure, Autonomy employer: Rivian Automotive
Contact Detail:
Rivian Automotive Recruiting Team
atcandidateaccommodations@rivian.com
StudySmarter Expert Advice 🤫
We think this is how you could land Senior Machine Learning Engineer, AI Infrastructure, Autonomy
✨Tip Number 1
Familiarise yourself with Rivian's mission and values. Understanding their commitment to sustainability and adventure will help you align your passion for the outdoors with their goals, making you a more appealing candidate.
✨Tip Number 2
Network with current or former employees of Rivian, especially those in similar roles. They can provide insights into the company culture and expectations, which can be invaluable during interviews.
✨Tip Number 3
Stay updated on the latest advancements in machine learning and AI infrastructure, particularly in the context of autonomous vehicles. Being knowledgeable about recent trends and technologies will demonstrate your commitment to the field.
✨Tip Number 4
Prepare to discuss specific projects where you've successfully implemented deep learning models. Be ready to explain your thought process, challenges faced, and how you overcame them, as this will showcase your problem-solving skills.
We think you need these skills to ace Senior Machine Learning Engineer, AI Infrastructure, Autonomy
Some tips for your application 🫡
Tailor Your CV: Make sure your CV highlights relevant experience in machine learning, particularly with deep learning models and frameworks like PyTorch. Emphasise any projects or roles that demonstrate your ability to design, train, and deploy large models.
Craft a Compelling Cover Letter: In your cover letter, express your passion for the outdoors and how it aligns with Rivian's mission. Discuss your technical skills and experiences that relate directly to the responsibilities outlined in the job description, such as your knowledge of transformer architecture and distributed training.
Showcase Relevant Projects: If you have worked on projects involving autonomous driving or edge computing systems, be sure to include these in your application. Detail your role, the technologies used, and the outcomes achieved to demonstrate your expertise.
Highlight Continuous Learning: Mention any ongoing education or certifications related to machine learning and AI infrastructure. This shows your commitment to staying updated in a rapidly evolving field, which is crucial for a role at Rivian.
How to prepare for a job interview at Rivian Automotive
✨Showcase Your Technical Expertise
Make sure to highlight your deep knowledge of PyTorch and any experience you have with Cuda or Triton. Be prepared to discuss specific projects where you've designed, trained, or deployed large deep learning models.
✨Demonstrate Problem-Solving Skills
Rivian values innovative thinking, so come ready to discuss how you've reframed old problems and sought new solutions in your previous roles. Share examples of how you've improved model training and inference speed.
✨Familiarise Yourself with the Company Mission
Understand Rivian's mission to keep the world adventurous and their commitment to sustainability. Be ready to express how your personal values align with theirs and how you can contribute to their goals.
✨Prepare for Technical Questions
Expect in-depth technical questions related to transformer architecture and distributed training. Brush up on model optimisation techniques like quantization and pruning, as these may come up during the interview.
