NVIDIA is seeking a Deep Learning Software Infrastructure Engineer to advance its Autonomous Vehicles project. The role focuses on building and scaling training libraries and infrastructure for end-to-end autonomous driving models. The engineer will work closely with research and platform teams to improve training efficiency and safety. This position offers the opportunity to make a significant impact in the field of AI and autonomous driving.

• 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, with a preference for PyTorch.
• Strong systems background in datacenter networking, parallel filesystems, storage systems, and schedulers.
• Proficiency in Python with experience writing production-grade libraries and automation tools.
• Ability to work closely with multi-functional teams and translate requirements into robust systems.

• Craft, scale, and harden deep learning infrastructure libraries and frameworks for training on multi-thousand GPU clusters.
• Improve efficiency throughout the training stack, including data loaders, distributed training, scheduling, and performance monitoring.
• Build robust training pipelines and libraries to handle massive video datasets and enable rapid experimentation.
• Collaborate with researchers, model engineers, and internal platform teams to enhance efficiency and improve training availability.
• Own core infrastructure components such as orchestration libraries and distributed training frameworks.
• Partner with leadership to ensure infrastructure scales with growing GPU capacity and dataset size while maintaining developer efficiency.

• Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.