Software Engineer, Pytorch Ecosystem, Machine Learning Infrastructure

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 library etc. 

We are looking for engineers to drive the PyTorch migration and build an ecosystem to support end-to-end development using PyTorch. Nuro’s current ML solutions are based on Tensorflow. Both the machine learning teams and ML Infra team are interested in migrating to Pytorch to catch up with the growing PyTorch community. You will be the one to work on a proof of concept of this effort and then set up a roadmap for the company-wide migration.

About the Work

• Architect the end-to-end ecosystem that supports machine learning development with PyTorch.
• Engineer distributed training platform on cloud and optimize computation resource efficiency.
• Collaborate with machine learning engineers to optimize model training and inference performance.
• Build frameworks and interfaces to unify machine learning development experience and accelerate machine learning evolution in Nuro.

About You

• 4+ years of work experience using the related technology or equivalent experience in Ph.D.
• You care for details and are able to lead the team in solving complex engineering problems.
• You have good experience using PyTorch. You have knowledge of the internals of deep learning frameworks and understand their implementation.
• You have good project management skills. You are able to drive the execution of major initiatives and identify blockers and risks.
• You have outstanding communication skills and enjoy working across organizations. You are able to align timelines and expectations with stakeholders.
• 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.