Find a job 
Featured Companies 
Featured Universities 
For Companies At a Glance
- Tasks: Join the Perception team to optimise ML training and inference for self-driving vehicles.
- Company: Rivian, a pioneering company on a mission to keep the world adventurous.
- Benefits: Competitive salary, inclusive culture, and opportunities for professional growth.
- Why this job: Make a real impact on the future of autonomous driving technology.
- Qualifications: PhD in relevant field and deep knowledge of PyTorch and ML infrastructure.
- Other info: Diverse team passionate about the outdoors and protecting the environment.
The predicted salary is between 36000 - 60000 £ per year.
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.
Role Summary
As a Staff Software Engineer, ML training and inference infrastructure, you will be a member of the Perception team at Rivian, which develops advanced machine learning algorithms that directly impact safety critical self-driving features of our category defining vehicles. We are looking for candidates with deep knowledge and strong enthusiasm towards establishing a state-of-art ML infrastructure for training and inference of large autonomous driving models; and optimizing the training and inference performance.
Responsibilities
- Optimize the performance of Deep Learning training workload on NVIDIA GPU systems on a large scale
- Optimize the latency of model inference and model pre- and post-processing on onboard systems
- Design, train, and deploy large deep learning models that can leverage the vast amount of labeled and unlabeled data
Qualifications
- PhD in CS/CE/EE, or equivalent, in industry experience
- Deep knowledge of PyTorch
- Knowledge of model training framework (e.g. PyTorch Lightning, ray, etc.)
- 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 models and doing detective work to improve model training and inference speed
Staff Software Engineer, ML Training and Inference Infrastructure employer: Rivian
Contact Detail:
Rivian Recruiting Team
StudySmarter Expert Advice 🤫
We think this is how you could land Staff Software Engineer, ML Training and Inference Infrastructure
✨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 that highlight your expertise in ML infrastructure, make sure to showcase them. A portfolio can speak volumes!
✨Tip Number 3
Prepare for the interview by diving deep into Rivian's mission and values. They love adventurous spirits, so be ready to discuss how your passion aligns with their goals.
✨Tip Number 4
Don't forget to apply through our website! It’s the best way to ensure your application gets the attention it deserves. Plus, we love seeing candidates who take that extra step.
We think you need these skills to ace Staff Software Engineer, ML Training and Inference Infrastructure
Some tips for your application 🫡
Show Your Passion: When writing your application, let your enthusiasm for machine learning and the outdoors shine through. We want to see that you’re not just qualified, but genuinely excited about the work we do at Rivian.
Tailor Your CV: Make sure your CV highlights relevant experience with deep learning and NVIDIA GPU systems. We love seeing candidates who can demonstrate their skills in optimising model performance and training workloads.
Be Clear and Concise: Keep your application straightforward and to the point. We appreciate clarity, so avoid jargon unless it’s necessary to showcase your expertise. Remember, we’re looking for someone who can communicate complex ideas simply.
Apply Through Our Website: Don’t forget to submit your application through our official website! It’s the best way for us to receive your details and ensures you’re considered for the role. Plus, it’s super easy!
How to prepare for a job interview at Rivian
✨Know Your Stuff
Make sure you brush up on your knowledge of PyTorch and transformer architectures. Be ready to discuss your experience with large-scale distributed training and how you've optimised model performance in the past. Rivian is looking for someone who can dive deep into technical details, so be prepared to showcase your expertise.
✨Show Your Passion for Adventure
Rivian values a love for the outdoors and a commitment to sustainability. Share your personal experiences that align with their mission. Whether it's a hiking trip or a project that involved eco-friendly tech, let them see your adventurous spirit and how it connects to their vision.
✨Prepare for Problem-Solving Questions
Expect to face some challenging scenarios during the interview. Rivian wants to know how you approach problem-solving, especially in unknown areas. Think of examples where you've reframed old problems or sought new solutions, and be ready to walk them through your thought process.
✨Ask Insightful Questions
At the end of the interview, don’t forget to ask questions that show your interest in the role and the company. Inquire about the team dynamics, the challenges they face in ML infrastructure, or how they measure success in their projects. This not only shows your enthusiasm but also helps you gauge if it's the right fit for you.
