Principal Prediction and Planning Machine Learning Engineer - Autonomous Vehicles

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

Job summary

An established industry player is seeking a Principal Prediction and Planning Machine Learning Engineer to join its Autonomous Vehicle team. This exciting role involves developing innovative machine learning solutions for autonomous driving platforms, focusing on prediction, planning, and control. Candidates will leverage their deep expertise in machine learning, particularly in deep neural networks and computer vision, to drive advancements in next-generation automotive products. Join a forward-thinking company at the forefront of AI automation and contribute to shaping the future of transportation.

Qualifications

• 15+ years of experience in building ML systems for autonomous vehicles.
• Strong understanding of large language models and transformers.

Responsibilities

• Research and implement deep-learning methods for prediction and planning.
• Collaborate with teams to deploy industrial-strength ML models.

Skills

Machine Learning

Deep Neural Networks

Computer Vision

Software Development

Analytical Skills

Interpersonal Skills

Communication Skills

Education

BS/MS/PhD in Computer Science

Electrical Engineering

Mechanical Engineering

Applied Math

Tools

PyTorch

TensorFlow

TensorRT

Job description

Principal Prediction and Planning Machine Learning Engineer - Autonomous Vehicles (Finance)

NVIDIA is seeking an experienced Machine Learning Engineer to join its Autonomous Vehicle team. You will develop key features for our autonomous driving platform, applying machine learning to prediction, planning, and control problems. The role involves creating innovative ML solutions for NVIDIA's next-generation automotive products. Candidates should have hands-on experience and deep knowledge in machine learning, with understanding of deep neural networks, autonomous vehicles, and domain adaptation being highly desirable. Collaboration across teams and expertise in computer vision, graphics, and machine learning are encouraged. Strong research, software development, communication, interpersonal, and analytical skills are essential. Join us to 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, run, and analyze experiments to evaluate solution efficiency on real-world data.
3. Partner with system software engineering teams to deploy industrial-strength ML models.
4. Collaborate with multi-functional teams through effective communication.

What we need to see:

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

Ways to stand out from the crowd:

• Experience as an ML planning lead.
• 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 across various domains including image classification, speech recognition, natural language processing, and autonomous vehicles. We are at the forefront of this AI era, developing hardware and software to lead the way. Join us and be part of the future!

The base salary range is $272,000 - $425,500, determined by location, experience, and current market rates. Additional benefits and equity are also provided. NVIDIA welcomes ongoing applications and is committed to diversity and equal opportunity in employment, promoting an inclusive environment free from discrimination based on race, religion, gender, age, veteran status, disability, or other protected characteristics.