Principal Prediction and Planning Machine Learning Engineer - Autonomous Vehicles

NVIDIA Corporation; Santa Clara (CA); On-site; USD 272,000 - 426,000; Full time

Job summary

Join a forward-thinking company at the forefront of AI innovation as a Machine Learning Engineer. In this pivotal role, you will develop cutting-edge solutions for autonomous vehicles, leveraging your expertise in machine learning and deep neural networks. Collaborate with multidisciplinary teams to create impactful ML systems that push the boundaries of technology. With a competitive salary and a commitment to diversity, this role offers a unique opportunity to shape the future of AI and automation. If you're passionate about driving innovation in the automotive industry, this position is for you.

Benefits

Equity

Comprehensive Benefits

Diversity Commitment

Qualifications

• 15+ years of experience in ML systems for autonomous vehicles.
• Strong understanding of LLMs and transformers.

Responsibilities

• Research and implement deep learning methods for autonomous vehicle products.
• Collaborate with teams to deploy ML models.

Skills

Machine Learning

Deep Neural Networks

Computer Vision

Software Development

Analytical Skills

Education

BS/MS/PhD in Computer Science

Electrical Engineering

Mechanical Engineering

Applied Math

Tools

PyTorch

TensorFlow

TensorRT

Job description

NVIDIA is seeking an experienced Machine Learning Engineer to join its Autonomous Vehicle team. In this role, you will develop key features for our autonomous driving platform by applying machine learning to prediction, planning, and control problems. You will create innovative ML solutions for NVIDIA's next-generation automotive products. The candidate should have hands-on experience and deep knowledge in machine learning, with a strong understanding of deep neural networks, autonomous vehicles, and domain adaptation being highly desirable. Collaboration across teams and disciplines such as computer vision, computer graphics, and machine learning is encouraged. Strong research, software development, communication, interpersonal, and analytical skills are essential. Join us and help shape the future of AI automation.

What you'll be doing:

1. Research, implement, and evaluate deep-learning-based methods for prediction and planning in NVIDIA's Autonomous Vehicle products.
2. Design, conduct, and analyze experiments to evaluate the efficiency of solutions on real-world data.
3. Collaborate with system software engineering specialists to deploy industrial-strength ML models.
4. Communicate and collaborate with multidisciplinary teams.

What we need to see:

• BS/MS/PhD in computer science, electrical engineering, mechanical engineering, applied math, or related fields (or equivalent experience).
• 15+ years of proven experience building ML systems for autonomous vehicles or similar robotics applications.
• Deep understanding of large language models (LLMs) and transformers.
• Hands-on experience building and deploying large-scale production ML systems.
• Excellent leadership skills and a track record of innovation in ML-based planning solutions.
• Experience with deep neural network training, inference, and optimization in frameworks like PyTorch, TensorFlow, TensorRT, etc.
• Strong understanding of the mathematical foundations of machine learning and deep learning.

Ways to stand out from the crowd:

• Prior experience as an ML planning lead.
• Proven publication record in ML for planning, vision, or related fields.
• Experience building and deploying vision-language action or chain-of-thought models.

At NVIDIA, we are powering a revolution in AI with deep learning techniques on NVIDIA GPUs, enabling breakthroughs in image classification, speech recognition, NLP, and autonomous vehicles. AI systems that learn, reason, and interact are no longer science fiction. Today, AI-powered self-driving cars can navigate country roads at night, and robots can learn motor skills through trial and error. We are at an exciting time in AI development. NVIDIA is leading this revolution with innovative hardware and software. Join us and be part of the future!

The base salary range is $272,000 - $425,500, determined by location, experience, and market factors. You will also be eligible for equity and benefits. NVIDIA accepts applications continuously.

We are committed to diversity and are an equal opportunity employer. We do not discriminate based on race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability, or any other protected characteristic.