Rivian
Senior Machine Learning Engineer, AI Infrastructure, Autonomy
How your CV stacks up
Upload your CV to see how well it fits this job role
?%
About 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.
Machine Learning Engineer – Mission-Critical
About the Role
We are looking for a full-time Machine Learning Engineer with deep knowledge and strong enthusiasm towards establishing a state-of-the-art AI infrastructure for training very large foundation models and accelerating model training/inference.
Our mission is to solve the autonomous driving problem. You will work with a team of talented software engineers, machine learning engineers, and research scientists to push the boundary of state-of-the-art machine learning models which will enable the next-generation end-to-end solution for autonomous driving.
Reasons to use Rodeo
Start with a chat, not a search bar
Grad scheme, placement, apprenticeship? Not sure what you want yet — that's fine. Your agent talks it through with you and turns "I have no idea" into a shortlist.
It searches the market for you
Every day your agent scans the market matching roles against what actually matters to you, not just keywords on a CV.
Only hits
No noise. No "maybe this fits." Just roles with a clear explanation of why they're right — and where to focus when applying.
Responsibilities
- Design, train, and deploy large deep learning models that can leverage vast quantities of labeled and unlabeled data from a fleet of millions of vehicles
- Improve the ecosystem for training infrastructure and deployment pipeline, accelerating model iteration and driving performance improvements
Qualifications
Required
- PhD in Computer Science (CS)/Computer Engineering (CE)/Electrical Engineering (EE), or equivalent industry experience
- Deep knowledge of PyTorch
- Experience with CUDA or Triton Language for developing custom operations
- Understanding of model training frameworks (e.g., PyTorch Lightning)
- In-depth knowledge of transformer architecture and optimizing training/inference of transformer models
- Experience with large-scale distributed training of large models
- Proven track record of model profiling and performance analysis to improve training/inference speed
Get help with your application
Your very own career expert that helps elevate your application to the next level.
Preferred
- Previous experience in the autonomous driving industry
- Knowledge of NVIDIA TensorRT
- Experience with edge computing systems
- Familiarity with model optimization techniques (e.g., quantization, pruning)
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 without regard to race, color, religion, national origin, ancestry, sex, sexual orientation, gender, gender expression, gender identity, genetic information, physical or mental disability, marital or domestic partner status, age, military/veteran status, medical condition, or any other characteristic protected by law.
Candidate Data Privacy
Please see official Rivian privacy policies for Candidate Personal Data handling, transfer, and retention procedures.
“It took my CV and asked me questions relevant to understanding what kind of jobs to suggest for me. Suggestions were almost perfect. Jobs were exactly what I’ve been looking for.”
Jessica, London
Skills
Location