NVIDIA is looking for a Machine Learning Engineer to build perception modules for autonomous vehicles that enable static world understanding without HD maps, ensuring scalability, adaptability, and safety across diverse road systems and corner cases.
Requirements
- 8+ years of experience applying ML/DL to real-world perception problems.
- Strong Python programming skills with proven software engineering practices.
- Hands-on experience with deep neural network training, inference, and optimization using PyTorch, TensorRT (or similar).
- Solid understanding of the mathematical foundations of ML/DL.
- Proven experience in data-driven analysis: setting up metrics, running large-scale experiments, interpreting results, and applying insights to guide model improvements.
- Experience developing scalable software for large, sophisticated systems.
- Proficiency in crafting perception systems that extend globally and adapt to different traffic environments.
- Strong background in data analytics, error analysis, and metric-driven iteration for ML systems.
- Proven track record to address rare and long-tail scenarios, from unusual road markings to sophisticated intersections.
- Hands-on work with lane detection, road boundaries, crosswalks, and similar tasks.
- Expertise with Transformers, BEV architectures, and modern static-world perception techniques.
Responsibilities
- Collaborate with researchers and engineers to transform innovative algorithms into production-ready perception modules.
- Research and prototype deep learning methods for lane graph construction, road boundary detection, traffic element recognition, and other static-world tasks.
- Develop and implement experiments at scale, applying detailed analytics to validate adaptability and improve generalization to rare and corner-case scenarios.
- Drive end-to-end deployment of perception models — from prototyping and validation to integration, optimization, and delivery into the autonomous driving stack.
Other
- MS or PhD in Computer Science, Engineering, or related field (focus on Deep Learning, AI, or similar), or equivalent experience.
- Excellent interpersonal skills; able to collaborate optimally across teams.
- Self-motivated, analytical, and eager to solve meaningful and challenging perception problems.
- Applications for this job will be accepted at least until September 30, 2025.
- NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer.
