Tech Lead, Machine Learning Infrastructure

Mountain View, California (HQ)

Applications have closed

Nuro, Inc.

Less driving. More thriving. Nuro autonomous, zero-occupant vehicles are making more possible—today.

View company page

Who We Are

Nuro exists to better everyday life through robotics. We have an elite team of entrepreneurs, engineers, designers, and scientists. We believe AI and robotics are at the cusp of transforming daily life and we are dedicated to building meaningful products with this technology. Join us and play a critical role in our mission.


About the Role

Our team is growing and we are looking for talented engineers to join us. Nuro is pursuing an ML-first software stack. Our team provides solutions to empower machine learning development in Nuro and optimize offboard training and onboard inference. Our solutions include a distributed training platform, TensorRT, Neural Architecture Search, model development toolbox etc. 

As a Tech Lead in ML Infra, you will contribute to the technical direction of our solutions in ML productivity and platform, working with many different technologies across the domains of machine learning, cloud computing and distributed systems. You will provide a vision of the scalable machine learning platform to support a growing community of machine learning in Nuro. You will work closely with other engineers in ML Infra to drive these initiatives from end-to-end.


About the Work

  • Architect the end-to-end ecosystem that supports machine learning development.
  • Design and engineer a distributed training platform on cloud and optimize computation resource efficiency.
  • Drive major initiatives to improve the tooling of machine learning development and troubleshooting.
  • Build frameworks and interfaces to unify machine learning development experience and accelerate machine learning evolution.


About You

  • 5+ years of work experience in deep learning or software infrastructure.
  • Have knowledge in using at least one deep learning framework, e.g. Tensorflow, Pytorch, Caffe. Be able to implement basic deep learning algorithms, e.g. classification, detection.
  • Have good knowledge in cloud computing and containerization technologies. You are able to architect distributed services using cloud infrastructure.
  • Good project management and communication skills. You are able to drive the execution of major initiatives and cross-team communication.
  • You are passionate about accelerating the benefits of robotics for everyday life. 

At Nuro, we celebrate differences and are committed to a diverse workplace that fosters inclusion and psychological safety for all employees. Nuro is proud to be an equal opportunity employer and expressly prohibits any form of workplace discrimination based on race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, genetic information, disability, veteran status, or any other legally protected characteristics.

You must be fully vaccinated against COVID-19 by your hire date to be eligible to start the role. Proof of vaccination will be required by your 

You must be fully vaccinated against COVID-19 by your hire date to be eligible to start the role. Proof of vaccination will be required by your start date. Nuro will consider and review accommodation requests based on medical, religious, or other grounds as required by applicable law for qualified candidates.

You must be fully vaccinated against COVID-19 by your hire date to be eligible to start the role. Proof of vaccination will be required by your start date. Nuro will consider and review accommodation requests based on medical, religious, or other grounds as required by applicable law for qualified candidates.

Tags: Architecture Caffe Classification Deep Learning Distributed Systems Machine Learning ML infrastructure ML models PyTorch Robotics TensorFlow TensorRT

Perks/benefits: Career development

Region: North America
Country: United States
Job stats:  5  0  0

More jobs like this

Explore more AI, ML, Data Science career opportunities

Find even more open roles in Artificial Intelligence (AI), Machine Learning (ML), Natural Language Processing (NLP), Computer Vision (CV), Data Engineering, Data Analytics, Big Data, and Data Science in general - ordered by popularity of job title or skills, toolset and products used - below.