Senior Deep Learning Engineer

Job Summary

To propel NVIDIA's Autonomous Vehicles project forward, the full-time Senior Deep Learning Software Infrastructure Engineer will build and scale training libraries and infrastructure for end-to-end autonomous driving models, enabling training on thousands of GPUs and massive datasets while collaborating with research and platform teams in a remote environment.

Key responsibilities

• Crafting, scaling, and hardening deep learning infrastructure libraries and frameworks for training on multi-thousand GPU clusters
• Improving efficiency throughout the training stack, including data loaders, distributed training, scheduling, and performance monitoring
• Collaborating with researchers and internal platform teams to enhance efficiency and improve training availability

Required qualifications

• BS, MS, or PhD in Computer Science, Electrical/Computer Engineering, or a related field, or equivalent experience
• 12+ years of professional experience building and scaling high-performance distributed systems, ideally in ML, HPC, or large-scale data infrastructure
• Extensive knowledge in deep learning frameworks, particularly PyTorch, and large-scale training techniques
• Strong systems background in datacenter networking, parallel filesystems, and schedulers
• Proficiency in Python with experience in writing production-grade libraries and automation tools