Find a job 
Featured Companies 
Featured Universities 
For Companies At a Glance
- Tasks: Design and deploy cutting-edge AI models for autonomous driving solutions.
- Company: Join Rivian, a leader in innovative automotive technology.
- Benefits: Competitive salary, inclusive culture, and opportunities for professional growth.
- Why this job: Be at the forefront of AI and make a real impact on the future of driving.
- Qualifications: PhD in relevant field and expertise in PyTorch and model training.
- Other info: Dynamic team environment with a commitment to diversity and inclusion.
The predicted salary is between 43200 - 72000 £ per year.
We are looking for a full-time Machine Learning Engineer, with deep knowledge and strong enthusiasm towards establishing a state-of-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-art machine learning models which will enable the next-generation E2E solution of autonomous driving.
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.
Senior Machine Learning Engineer, AI Infrastructure, Autonomy employer: Rivian
Contact Detail:
Rivian Recruiting Team
StudySmarter Expert Advice 🤫
We think this is how you could land Senior Machine Learning Engineer, AI Infrastructure, Autonomy
✨Tip Number 1
Network like a pro! Reach out to folks in the industry, especially those at Rivian. A friendly chat can open doors and give you insights that a job description just can't.
✨Tip Number 2
Show off your skills! If you've got projects or contributions to open-source that highlight your expertise in machine learning, make sure to showcase them. A portfolio speaks volumes!
✨Tip Number 3
Prepare for the interview by brushing up on your technical knowledge. Be ready to discuss your experience with PyTorch, model training frameworks, and any relevant projects. Confidence is key!
✨Tip Number 4
Don't forget to apply through our website! It’s the best way to ensure your application gets seen. Plus, it shows you're genuinely interested in joining the team at Rivian.
We think you need these skills to ace Senior Machine Learning Engineer, AI Infrastructure, Autonomy
Some tips for your application 🫡
Show Your Passion: When writing your application, let your enthusiasm for AI and machine learning shine through. We want to see that you’re not just qualified, but genuinely excited about the role and our mission in autonomous driving.
Tailor Your CV: Make sure your CV highlights relevant experience and skills that match the job description. We love seeing how your background aligns with what we’re looking for, so don’t be shy about showcasing your expertise in PyTorch and model training!
Craft a Compelling Cover Letter: Your cover letter is your chance to tell us why you’re the perfect fit for this role. Use it to explain your experience with large-scale distributed training and any projects that demonstrate your problem-solving skills in machine learning.
Apply Through Our Website: We encourage you to apply directly through our website. It’s the best way for us to receive your application and ensures you’re considered for the role. Plus, it makes the whole process smoother for everyone involved!
How to prepare for a job interview at Rivian
✨Know Your Tech Inside Out
Make sure you’re well-versed in the technologies mentioned in the job description, especially PyTorch and CUDA. Brush up on your knowledge of transformer architectures and model optimisation techniques like quantisation and pruning. Being able to discuss these topics confidently will show that you’re not just familiar with them, but that you can apply them effectively.
✨Showcase Your Problem-Solving Skills
Prepare to discuss specific challenges you've faced in previous roles, particularly around model training and inference speed. Use examples that highlight your detective work in profiling models and improving performance. This will demonstrate your hands-on experience and analytical thinking, which are crucial for this role.
✨Understand the Autonomous Driving Landscape
Familiarise yourself with the current trends and challenges in the autonomous driving industry. Be ready to discuss how your skills can contribute to solving these problems. Showing that you have a genuine interest in the field will set you apart from other candidates.
✨Ask Insightful Questions
Prepare thoughtful questions about the team’s current projects, the AI infrastructure they’re building, and their vision for the future. This not only shows your enthusiasm for the role but also helps you gauge if the company aligns with your career goals. Engaging in a two-way conversation can leave a lasting impression.
