Senior ML Engineer, Planning and Prediction – Autonomous Vehicles

Apply Now

Job Description

We are seeking an energetic, hardworking machine learning engineer to develop key features for our autonomous driving platform. In this role, you will apply machine learning to a mix of planning, prediction and control problems. You will develop innovative ML solutions for NVIDIA’s next-generation automotive products. Join us and help craft the future of AI automation.

What you'll be doing:

  • You will research, implement and evaluate deep-learning-based methods for planning and prediction
  • You will design, run, and analyze experiments to evaluate the efficiency of our solutions on real-world data.
  • Partner with system software engineering specialists to ship industrial strength ML models

What we need to see:

  • MS or PhD level education in Engineering or Computer Science with a focus on Deep Learning, Artificial Intelligence, or a related field, or equivalent experience.
  • Excellent written and verbal communication skills and strong interpersonal skills.
  • 5+ years of practical experience applying machine learning to address real-world problems
  • Strong Python programming experience with software design skills.
  • Experience with deep neural network training, inference and optimization in leading frameworks (e.g. Pytorch, Tensorflow, TensorRT)
  • Practical software engineering experience, developing for large and complex systems.
  • Excellent understanding of the mathematical foundations of machine learning and deep learning
  • Self-motivation combined with team spirit and the urge to learn new things and tackle hard problems.

Ways to stand out from the crowd:

We definitely want to hear from you if you are an upbeat contributor with a background that includes one or more of the following:

  • Experience in applying deep learning to prediction and planning problems for autonomous vehicles
  • Experience with techniques in reinforcement learning or optimal-control
  • Experience with GPGPU programming (CUDA)
  • Publications at international relevant conferences (CVPR, NeurIPS, ICRA, etc)

Academic and commercial groups around the world are powering a revolution in artificial intelligence using deep learning techniques running on NVIDIA GPUs, enabling breakthroughs in problems from image classification to speech recognition to natural language processing and autonomous vehicles. Intelligent AI computers that can learn, reason and interact with people are no longer science fiction. Today, a self-driving car powered by AI can meander through a country road at night and find its way. An AI-powered robot can learn motor skills through trial and error. This is truly an outstanding time. The era of AI has begun and NVIDIA is leading the drive with revolutionary hardware and software. Come join us at NVIDIA!

Company Info.


NVIDIA’s invention of the GPU sparked the PC gaming market. The company’s pioneering work in accelerated computing—a supercharged form of computing at the intersection of computer graphics, high performance computing and AI—is reshaping trillion-dollar industries, such as transportation, healthcare and manufacturing, and fueling the growth of many others.

  • Industry
    Cloud computing,Video games,Computer software,Semiconductors,Computer hardware,Consumer electronics,Artificial intelligence
  • No. of Employees
  • Location
    2701 San Tomas Expressway, Santa Clara, CA 95050, USA
  • Website
  • Jobs Posted

Get Similar Jobs In Your Inbox

NVIDIA is currently hiring Senior Machine Learning Engineer Jobs in Beijing, China with average base salary of ¥300,500 - ¥650,500 / Year.

Similar Jobs View More