Machine Learning Infrastructure Technical Lead Manager, Nuro
$202-303k
+ Annual performance bonus + Equity
Office located in Mountain View, CA
Autonomous delivery vehicle manufacturer
Be an early applicant
Autonomous delivery vehicle manufacturer
501-1000 employees
Be an early applicant
$202-303k
+ Annual performance bonus + Equity
Office located in Mountain View, CA
501-1000 employees
Company mission
To better everyday life through robotics.
Role
Who you are
- 6+ years of industry or research experience in ML Infra, distributed training, model inference, or GPU programming
- Strong expertise in deep learning frameworks (PyTorch, TensorFlow, JAX) and algorithms across CV, NLP, and planning
- Advanced programming skills in C++/C and Python
- Excellent communication skills with proven ability to drive technical innovation
Desirable
- Experience with CUDA, Cublas, Cudnn or any other Nvidia SDKs
- Experience with model quantization or pruning
- Experience with compilers or ML compilers (e.g. TensorRT, Triton, XLA, Clang)
- Experience with AI algorithms and hardware codesign (e.g. Depthwise Conv, Flash Attention)
- Experience with distributed training speedup (e.g. FSDP, DeepSpeed)
What the job involves
- The ML Infra team is growing and we are looking for talented leaders to join us. To support the ML-first software stack of Nuro, our team provides solutions to empower machine learning development in Nuro and optimize on-cloud training and onboard inference
- Our solutions include a distributed training platform, ML compiler, model components libraries etc
- We are looking for a tech lead manager to grow the team, lead strategic vision and ML infra team, cross-team alignments with stakeholders, and drive key technology advancement
- Identify and resolve critical bottlenecks in the lifecycle of ML model development and deployment
- Spearhead research and implementation of advanced ML Infra solutions
- Develop and execute technical roadmaps
- Drive adoption of team solutions across ML teams
- Foster engineering excellence and continuous improvement
- Provide technical mentorship and drive professional development of team members
Our take
Nuro offers custom-built autonomous vehicles for the delivery of physical goods, rather than people. Its vehicles have a distinctive, futuristic, design that has not only made the vehicles noticeable from a marketing point of view, but, has also made its vehicles stand out to other road users, improving safety.
Nuro is an exciting addition to the autonomous vehicle landscape. Without a passenger inside the vehicle, it has been able to sidestep some of the safety issues that have plagued other autonomous vehicle brands. Additionally, the company has landed several high-profile partners, including Uber, Walmart, Chipotle, and Kroger.
With home deliveries at an all-time high, many companies are looking for ways to get goods to customers’ homes as fast as possible. Low-cost, efficient services such as Nuro's offering are becoming increasingly attractive. The company is turning its focus to commercializing and scaling the production of its third-generation vehicle at a new facility in Southern Nevada as it hopes to expand its services across the US.
Steph
Company Specialist at Welcome to the Jungle
Insights
Company
Funding (last 2 of 5 rounds)
Nov 2021
$600m
SERIES D
Nov 2020
$500m
SERIES C
Total funding: $2.1bn
Company benefits
- Free Caltrain pass and commuter benefits
- Company stock options
- Work from home opportunities
- Health insurance
Company values
- Earn it - We seek and expect excellence and are not afraid of hard work
- 1% better everyday - We continuously focus on improving both as individuals and as a company
- Find a way - We embrace uncertainty and thrive on solving the hardest problems
- Act now - We trust our team's judgment—everyone is a problem solver
- One team - Nuro only succeeds if we are all working toward a common goal
Company HQ
Mountain View, CA
Articles
Leadership
Jiajun Zhu
(CEO)After an MA in Computer Science at the University of Virginia, Jiajun worked as Principal Software Engineer at Google for over 8 years. He left this role in August 2016 to co-found Nuro with Dave.
Dave Ferguson
(President)Experienced Robotics Engineer with stints at Google (Computer Vision and Machine Learning Lead) and Intel (Senior Research Scientist). PhD in Robotics from Carnegie Mellon University.
Salary benchmarks
We don't have enough data yet to provide salary benchmarks for this role.
Submit your salary to help other candidates with crowdsourced salary estimates.
Share this job
View 42 more jobs at Nuro