Senior Machine Learning Engineer

NVIDIA offers an onsite opportunity in Santa Clara, CA to help build the machine learning backbone for DRIVE AV perception. The role sits at the intersection of LiDAR and camera perception, data pipelines, and production-ready C++ code, with a competitive salary range of USD 184,000 to 356,500 per year, plus equity and comprehensive benefits. You will collaborate with LiDAR and camera teams to deliver reliable autonomous driving solutions and advance perception technology in a production environment.

Responsibilities

• Model Development: Design, train, and optimize innovative ML models for LiDAR perception, including road element detection, semantic segmentation, and tracking.
• End-to-end ML workflows: Develop and coordinate data pipelines, model training, metrics, continuous performance instrumentation, and reporting.
• Productization: Transition ML models from evaluation to shipping as part of the NVIDIA DRIVE AV platform, writing highly efficient production code in C++.
• Innovation: Stay current with ML advancements and integrate techniques that boost platform performance.
• Collaboration: Work with LiDAR and camera teams, developers, engineers, and managers to turn complex ideas into reliable autonomous driving solutions.

Requirements

• BS or MS in Computer Science, Engineering, or a related field, or equivalent experience.
• 6+ years of relevant industry experience applying machine learning to real-world problems.
• Strong C++ and Python programming and debugging skills for large, complex systems.
• Deep practical experience applying ML to LiDAR/camera perception in automotive or related fields.
• Experience with deep learning frameworks such as PyTorch and TensorFlow, with solid understanding of ML fundamentals.
• Experience building and sustaining training and essential metric workflows for large-scale datasets.
• Excellent communication and analytical skills, with self-motivated drive to solve hard problems.

Technologies

• C++, Python
• PyTorch
• TensorFlow
• TensorRT

Benefits

• Equity
• Benefits

Ways to stand out from the crowd

• LiDAR or Camera Perception Experience: Proven track record of developing and shipping deep learning models for LiDAR/Camera in a production environment.
• Advanced Model Knowledge: Familiarity with modern architectures such as Transformers and their application to visual recognition tasks.
• AV Production Experience: History of delivering ML features and models into a production autonomous vehicle stack or robotics product.
• Performance Optimization: Experience with model optimization for real-time inference on embedded or automotive platforms, including TensorRT.