Senior Machine Learning Engineer

Details

Intelligent machines powered by artificial intelligence can learn, reason, and interact with people. NVIDIA's GPU runs deep learning algorithms that simulate human intelligence and powers computers, robots, and self-driving vehicles.

The team is building the machine learning backbone for the Perception component of NVIDIA DRIVE AV. The role focuses on computer vision, LiDAR and camera perception, and AI infrastructure for autonomous driving in unconstrained environments.

Responsibilities

• Design, train, and optimize machine learning models for LiDAR perception, including road element detection, semantic segmentation, and tracking.
• Develop and coordinate complete machine learning workflows covering data pipelines, model training, model metrics, continuous performance instrumentation, and reporting.
• Productize machine learning models and algorithms, taking them from evaluation and experimentation through shipping as part of the NVIDIA DRIVE AV platform.
• Develop highly efficient product code in C++.
• Track developments in machine learning and incorporate techniques that improve platform performance.
• Collaborate with LiDAR and camera teams, developers, engineers, and managers to turn complex ideas into reliable autonomous-driving solutions.

Requirements

• Bachelor's or master's degree in Computer Science, Engineering, or a related field, or equivalent experience.
• Six or more years of relevant industry experience applying machine learning to real-world problems.
• Strong C++ and Python programming and debugging skills, with experience developing for large, complex systems.
• Deep practical experience applying machine learning to LiDAR or camera perception in automotive or related fields.
• Experience with deep learning frameworks such as PyTorch and TensorFlow.
• Strong understanding of the mathematical foundations of machine learning.
• Experience building and sustaining training and essential metric workflows for large-scale datasets.
• Excellent communication and analytical skills, with a self-motivated approach to solving difficult problems.

Preferred Qualifications

• Proven experience developing and shipping deep learning models for LiDAR or camera perception in a production environment.
• Familiarity with modern network architectures such as Transformers and their application to visual recognition tasks.
• Experience delivering machine learning features and models into a production autonomous vehicle stack or related robotics product.
• Experience optimizing models for real-time inference on embedded or automotive platforms, such as with TensorRT.

Compensation and Benefits

The base salary is determined by location, experience, and the pay of employees in similar positions. The base salary range is USD 184,000–287,500 for Level 4 and USD 224,000–356,500 for Level 5. The role also includes eligibility for equity and benefits.

Applications will be accepted at least until August 9, 2026. This posting is for an existing vacancy. NVIDIA uses AI tools in its recruiting processes and is an equal opportunity employer.