Staff Software Engineer, ML Training and Inference Infrastructure Careers at Rivian - London, London | Apply Now!
Rivian
Staff Software Engineer, ML Training and Inference Infrastructure Careers at Rivian - London, London | Apply Now!
Job Description
Role Overview
Join Rivian's revolutionary Perception team as a Staff Software Engineer, ML Training and Inference Infrastructure in London, UK. This is your chance to shape the future of autonomous driving technology at one of the world's most innovative electric vehicle manufacturers. Rivian is building emissions-free Electric Adventure Vehicles with cutting-edge self-driving capabilities, and we're seeking world-class ML engineers to optimize our massive deep learning infrastructure.
As a key member of our London team, you'll tackle the most challenging problems in large-scale ML training and inference optimization. Working with NVIDIA GPU clusters and onboard vehicle systems, you'll push the boundaries of what's possible with transformer models and distributed training frameworks. If you have deep PyTorch expertise and a passion for performance optimization, this role will position you at the forefront of automotive AI innovation.
Rivian's mission to keep the world adventurous forever means we're not just building cars—we're creating intelligent systems that will redefine mobility. Your work will directly impact safety-critical self-driving features that millions will rely on daily.
Key Responsibilities at Rivian
Take ownership of Rivian's ML infrastructure stack with these critical responsibilities:
- Optimize Deep Learning training workloads across large-scale NVIDIA GPU systems for maximum throughput
- Minimize model inference latency on resource-constrained onboard vehicle hardware
- Design and deploy massive deep learning models leveraging petabytes of labeled/unlabeled driving data
- Implement distributed training strategies using PyTorch Lightning, Ray, and custom frameworks
- Profile and accelerate transformer architectures for both training and inference phases
- Build state-of-the-art ML infrastructure supporting Rivian's autonomous driving roadmap
- Conduct deep performance analysis and detective work to eliminate training bottlenecks
- Collaborate with Perception team on safety-critical feature deployment
- Optimize pre- and post-processing pipelines for real-time inference performance
Qualifications & Requirements
Rivian seeks candidates who bring exceptional technical depth and proven impact:
- PhD in CS/CE/EE or equivalent industry experience building ML systems at scale
- Deep PyTorch mastery including advanced features and custom extensions
- Expertise in training frameworks (PyTorch Lightning, Ray, DeepSpeed, etc.)
- Transformer architecture specialist with proven acceleration techniques
- Distributed training veteran with experience scaling to hundreds of GPUs
- Performance profiling expert who loves solving complex optimization puzzles
Bonus points for experience with autonomous driving datasets, NVIDIA CUDA optimization, and production ML deployment at automotive scale.
Salary & Benefits
Competitive Compensation: £140,000 - £180,000 base salary (estimated) plus performance bonuses, equity, and comprehensive benefits package tailored for senior ML talent in London.
- Market-leading base salary with annual performance bonuses
- Significant Rivian stock options and RSUs
- Private medical, dental, and vision insurance
- 25+ days holiday plus UK bank holidays
- Hybrid work model with London office flexibility
- Rivian vehicle purchase/lease programs
- Professional development stipend (£3,000+ annually)
- Enhanced parental leave and family benefits
- Sustainable commuting allowances
- Mental health support and wellbeing programs
Why Join Rivian?
Rivian isn't just another EV company—we're redefining adventure through intelligent, sustainable mobility. Here's why top ML talent chooses Rivian:
- Impact at Scale: Your optimizations will power millions of autonomous miles
- Cutting-Edge Tech: Latest NVIDIA GPUs, petabyte-scale datasets, transformer research
- Adventure Culture: Team offsites, outdoor activities, EV test drives
- Global Reach: London hub connecting US innovation with European talent
- Mission-Driven: Protecting the outdoors through emissions-free adventure vehicles
Join a diverse team of curious, courageous engineers who challenge conventional thinking and deliver category-defining results. Rivian provides equal opportunity employment and is committed to accessibility for all candidates.
How to Apply
Ready to accelerate the future of autonomous driving? Applications close soon—don't miss your chance to join Rivian's London Perception team:
- Submit your CV highlighting PyTorch/distributed training experience
- Include GitHub/portfolio links to ML optimization projects
- Email candidateaccommodations@rivian.com for accessibility needs
- No third-party applications accepted
- Direct applications only through Rivian careers portal
Apply now and help Rivian keep the world adventurous forever through breakthrough ML infrastructure!
Locations
- London, London, United Kingdom
Salary
Skills Required
- PyTorchintermediate
- Deep Learningintermediate
- NVIDIA GPU Optimizationintermediate
- Transformer Modelsintermediate
- Distributed Trainingintermediate
- Model Inferenceintermediate
- Machine Learning Infrastructureintermediate
- Autonomous Drivingintermediate
- Performance Profilingintermediate
- PyTorch Lightningintermediate
- Ray Frameworkintermediate
- Large Scale ML Trainingintermediate
- GPU Accelerationintermediate
- Model Optimizationintermediate
- Self-Driving AIintermediate
Required Qualifications
- PhD in Computer Science, Computer Engineering, Electrical Engineering, or equivalent industry experience (experience)
- Deep knowledge of PyTorch framework and its advanced features (experience)
- Expertise in model training frameworks like PyTorch Lightning, Ray, or similar (experience)
- In-depth understanding of transformer architecture and acceleration techniques (experience)
- Proven experience with large-scale distributed training of deep learning models (experience)
- Strong track record of profiling models and optimizing training/inference speed (experience)
Responsibilities
- Optimize Deep Learning training workloads on large-scale NVIDIA GPU systems
- Reduce model inference latency and optimize pre/post-processing on onboard vehicle systems
- Design, train, and deploy large-scale deep learning models for autonomous driving
- Leverage vast labeled and unlabeled data for model development
- Establish state-of-the-art ML infrastructure for training and inference
- Profile and debug models to achieve maximum training and inference performance
- Implement distributed training strategies for massive autonomous driving datasets
- Collaborate with Perception team on safety-critical self-driving features
- Accelerate transformer model training and inference performance
Benefits
- general: Competitive salary with performance bonuses
- general: Comprehensive health and dental insurance
- general: Generous stock options and equity packages
- general: 25+ days annual leave plus bank holidays
- general: Hybrid work flexibility in London office
- general: State-of-the-art EV testing facilities access
- general: Professional development budget and conferences
- general: Rivian employee vehicle purchase program
- general: Sustainable transportation benefits
- general: Team adventure and outdoor activities
- general: Enhanced parental leave policies
- general: Mental health and wellbeing programs
Tags & Categories
Answer 10 quick questions to check your fit for Staff Software Engineer, ML Training and Inference Infrastructure Careers at Rivian - London, London | Apply Now! @ Rivian.
Related Books and Jobs
Staff Software Engineer, ML Training and Inference Infrastructure Careers at Rivian - London, London | Apply Now!
Rivian
Staff Software Engineer, ML Training and Inference Infrastructure Careers at Rivian - London, London | Apply Now!
Job Description
Role Overview
Join Rivian's revolutionary Perception team as a Staff Software Engineer, ML Training and Inference Infrastructure in London, UK. This is your chance to shape the future of autonomous driving technology at one of the world's most innovative electric vehicle manufacturers. Rivian is building emissions-free Electric Adventure Vehicles with cutting-edge self-driving capabilities, and we're seeking world-class ML engineers to optimize our massive deep learning infrastructure.
As a key member of our London team, you'll tackle the most challenging problems in large-scale ML training and inference optimization. Working with NVIDIA GPU clusters and onboard vehicle systems, you'll push the boundaries of what's possible with transformer models and distributed training frameworks. If you have deep PyTorch expertise and a passion for performance optimization, this role will position you at the forefront of automotive AI innovation.
Rivian's mission to keep the world adventurous forever means we're not just building cars—we're creating intelligent systems that will redefine mobility. Your work will directly impact safety-critical self-driving features that millions will rely on daily.
Key Responsibilities at Rivian
Take ownership of Rivian's ML infrastructure stack with these critical responsibilities:
- Optimize Deep Learning training workloads across large-scale NVIDIA GPU systems for maximum throughput
- Minimize model inference latency on resource-constrained onboard vehicle hardware
- Design and deploy massive deep learning models leveraging petabytes of labeled/unlabeled driving data
- Implement distributed training strategies using PyTorch Lightning, Ray, and custom frameworks
- Profile and accelerate transformer architectures for both training and inference phases
- Build state-of-the-art ML infrastructure supporting Rivian's autonomous driving roadmap
- Conduct deep performance analysis and detective work to eliminate training bottlenecks
- Collaborate with Perception team on safety-critical feature deployment
- Optimize pre- and post-processing pipelines for real-time inference performance
Qualifications & Requirements
Rivian seeks candidates who bring exceptional technical depth and proven impact:
- PhD in CS/CE/EE or equivalent industry experience building ML systems at scale
- Deep PyTorch mastery including advanced features and custom extensions
- Expertise in training frameworks (PyTorch Lightning, Ray, DeepSpeed, etc.)
- Transformer architecture specialist with proven acceleration techniques
- Distributed training veteran with experience scaling to hundreds of GPUs
- Performance profiling expert who loves solving complex optimization puzzles
Bonus points for experience with autonomous driving datasets, NVIDIA CUDA optimization, and production ML deployment at automotive scale.
Salary & Benefits
Competitive Compensation: £140,000 - £180,000 base salary (estimated) plus performance bonuses, equity, and comprehensive benefits package tailored for senior ML talent in London.
- Market-leading base salary with annual performance bonuses
- Significant Rivian stock options and RSUs
- Private medical, dental, and vision insurance
- 25+ days holiday plus UK bank holidays
- Hybrid work model with London office flexibility
- Rivian vehicle purchase/lease programs
- Professional development stipend (£3,000+ annually)
- Enhanced parental leave and family benefits
- Sustainable commuting allowances
- Mental health support and wellbeing programs
Why Join Rivian?
Rivian isn't just another EV company—we're redefining adventure through intelligent, sustainable mobility. Here's why top ML talent chooses Rivian:
- Impact at Scale: Your optimizations will power millions of autonomous miles
- Cutting-Edge Tech: Latest NVIDIA GPUs, petabyte-scale datasets, transformer research
- Adventure Culture: Team offsites, outdoor activities, EV test drives
- Global Reach: London hub connecting US innovation with European talent
- Mission-Driven: Protecting the outdoors through emissions-free adventure vehicles
Join a diverse team of curious, courageous engineers who challenge conventional thinking and deliver category-defining results. Rivian provides equal opportunity employment and is committed to accessibility for all candidates.
How to Apply
Ready to accelerate the future of autonomous driving? Applications close soon—don't miss your chance to join Rivian's London Perception team:
- Submit your CV highlighting PyTorch/distributed training experience
- Include GitHub/portfolio links to ML optimization projects
- Email candidateaccommodations@rivian.com for accessibility needs
- No third-party applications accepted
- Direct applications only through Rivian careers portal
Apply now and help Rivian keep the world adventurous forever through breakthrough ML infrastructure!
Locations
- London, London, United Kingdom
Salary
Skills Required
- PyTorchintermediate
- Deep Learningintermediate
- NVIDIA GPU Optimizationintermediate
- Transformer Modelsintermediate
- Distributed Trainingintermediate
- Model Inferenceintermediate
- Machine Learning Infrastructureintermediate
- Autonomous Drivingintermediate
- Performance Profilingintermediate
- PyTorch Lightningintermediate
- Ray Frameworkintermediate
- Large Scale ML Trainingintermediate
- GPU Accelerationintermediate
- Model Optimizationintermediate
- Self-Driving AIintermediate
Required Qualifications
- PhD in Computer Science, Computer Engineering, Electrical Engineering, or equivalent industry experience (experience)
- Deep knowledge of PyTorch framework and its advanced features (experience)
- Expertise in model training frameworks like PyTorch Lightning, Ray, or similar (experience)
- In-depth understanding of transformer architecture and acceleration techniques (experience)
- Proven experience with large-scale distributed training of deep learning models (experience)
- Strong track record of profiling models and optimizing training/inference speed (experience)
Responsibilities
- Optimize Deep Learning training workloads on large-scale NVIDIA GPU systems
- Reduce model inference latency and optimize pre/post-processing on onboard vehicle systems
- Design, train, and deploy large-scale deep learning models for autonomous driving
- Leverage vast labeled and unlabeled data for model development
- Establish state-of-the-art ML infrastructure for training and inference
- Profile and debug models to achieve maximum training and inference performance
- Implement distributed training strategies for massive autonomous driving datasets
- Collaborate with Perception team on safety-critical self-driving features
- Accelerate transformer model training and inference performance
Benefits
- general: Competitive salary with performance bonuses
- general: Comprehensive health and dental insurance
- general: Generous stock options and equity packages
- general: 25+ days annual leave plus bank holidays
- general: Hybrid work flexibility in London office
- general: State-of-the-art EV testing facilities access
- general: Professional development budget and conferences
- general: Rivian employee vehicle purchase program
- general: Sustainable transportation benefits
- general: Team adventure and outdoor activities
- general: Enhanced parental leave policies
- general: Mental health and wellbeing programs
Tags & Categories
Answer 10 quick questions to check your fit for Staff Software Engineer, ML Training and Inference Infrastructure Careers at Rivian - London, London | Apply Now! @ Rivian.
Related Books and Jobs