Software Engineer, Ph.D Intern - Machine Learning Infrastructure

Who We Are

Nuro exists to better everyday life through robotics. The company’s custom electric autonomous vehicles are designed to bring the things you need—from produce to prescriptions—right to your home. Nuro’s autonomous, goods-focused solution can give you valuable time back and more freedom to do what you love. This convenient, eco-friendly alternative to driving has the potential to make streets safer and cities more livable.

About the Work

Our software organization is growing and we are looking for talented students to join our Autonomy Platform, Machine Learning Infra team as part of the Pathfinders Internship Program. 

As a Pathfinder, you will gain meaningful work experience while applying the knowledge and skills you have gained in the classroom. You will have the opportunity to make connections with interns and employees across the company, and to make a quantifiable impact on one of our teams. 

We hire students at the undergraduate and graduate levels for roles lasting 12, 14 or 16 weeks. Longer term internships are available for research positions.

As an ML Infra Engineer Intern, you will work very closely with infrastructure and machine learning teams on the following topics:

• Optimize onboard machine learning model latency and memory usage using one or more of the following: deep learning graph compression; TensorRT; CUDA programming; deep learning compilers.
• Engineer tools to profile and monitor model performance, to debug model inference problems.
• Collaborate with ML engineers to understand AI problems, implement solutions, and deliver efficient models.

About You

• You are pursuing a Ph.D degree in computer science or related field.
• You have experience in deep learning and deep learning frameworks such as Tensorflow and Pytorch.
• You have experience in one or more of the following areas: CUDA programming; TensorRT, model optimization, model compression; deep learning compilers.
• You have strong problem solving and programming skills with advanced C++ and Python skills.

For this position, the reasonably expected pay range is between $11,500 and $14,000/month. Your actual base pay will depend on several factors, including your experience, qualifications, education, location, and skills.  

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.