Find a job 
Featured Companies 
Featured Universities 
For Companies At a Glance
- Tasks: Optimise deep learning workloads and deploy large models on NVIDIA GPU systems.
- Company: Rivian builds emissions-free Electric Adventure Vehicles and values diverse, adventurous teams.
- Benefits: Enjoy a collaborative culture with opportunities for remote work and personal growth.
- Why this job: Join a mission-driven team that challenges the status quo and protects the environment.
- Qualifications: PhD in relevant fields with expertise in PyTorch and large-scale model training.
- Other info: Rivian is committed to equal opportunity and accessible hiring processes.
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
- 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 Automotive
Contact Detail:
Rivian Automotive Recruiting Team
atcandidateaccommodations@rivian.com
StudySmarter Expert Advice 🤫
We think this is how you could land Staff Software Engineer, ML Training and Inference Infrastructure
✨Tip Number 1
Familiarise yourself with Rivian's mission and values. Understanding their commitment to adventure and sustainability will help you align your passion for the outdoors with their goals, making you a more appealing candidate.
✨Tip Number 2
Showcase your experience with NVIDIA GPU systems and deep learning frameworks like PyTorch. Be prepared to discuss specific projects where you've optimised performance or reduced latency, as this will demonstrate your technical expertise.
✨Tip Number 3
Network with current or former Rivian employees on platforms like LinkedIn. Engaging in conversations about their experiences can provide valuable insights into the company culture and may even lead to referrals.
✨Tip Number 4
Prepare to discuss your approach to problem-solving, especially in areas of model training and inference. Rivian values innovative thinking, so be ready to share examples of how you've tackled complex challenges in your previous roles.
We think you need these skills to ace Staff Software Engineer, ML Training and Inference Infrastructure
Some tips for your application 🫡
Understand the Role: Before applying, make sure you fully understand the responsibilities and qualifications for the Staff Software Engineer position. Familiarise yourself with the specific technologies mentioned, such as PyTorch and transformer architecture.
Tailor Your CV: Customise your CV to highlight relevant experience in deep learning, model training, and optimisation. Emphasise any projects or roles where you've worked with NVIDIA GPU systems or large-scale distributed training.
Craft a Compelling Cover Letter: Write a cover letter that reflects your passion for the outdoors and aligns with Rivian's mission. Discuss how your background and skills can contribute to their innovative approach in the electric vehicle space.
Showcase Your Achievements: In your application, include specific examples of past successes in optimising model performance or improving inference speed. Use metrics to quantify your impact wherever possible.
How to prepare for a job interview at Rivian Automotive
✨Showcase Your Technical Expertise
Be prepared to discuss your deep knowledge of PyTorch and any model training frameworks you've used. Highlight specific projects where you've optimised performance or reduced latency, especially in large-scale distributed training.
✨Demonstrate Problem-Solving Skills
Rivian values innovative thinking. Be ready to share examples of how you've reframed old problems and sought new solutions in your previous roles, particularly in the context of machine learning and deep learning models.
✨Understand Rivian's Mission
Familiarise yourself with Rivian's commitment to sustainability and adventure. Show your passion for the outdoors and how it aligns with their mission, as cultural fit is just as important as technical skills.
✨Prepare for Behavioural Questions
Expect questions that assess your teamwork and collaboration skills. Prepare examples that demonstrate how you've worked effectively in diverse teams and contributed to a positive work environment.
