Senior Deep Learning Systems Engineer, Autonomous Vehicles

Socket.dev; Colorado; On-site; USD 224,000 - 357,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is seeking a Senior Deep Learning Systems Engineer to advance its Autonomous Vehicles project. You will build and scale training libraries and infrastructure to enable end-to-end autonomous driving models on thousands of GPUs, collaborating with research and platform teams to boost efficiency and availability.

Ideal candidates have 12+ years with deep learning systems, strong Python/C++, PyTorch expertise, and a solid systems background in data centers, networking, and orchestration

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 (PyTorch preferred), large scale training (DDP/FSDP, NCCL, tensor/pipeline parallelism), and performance profiling.
• Strong systems background: datacenter networking (RoCE, IB), parallel filesystems (Lustre), storage systems, schedulers (Slurm, Kubernetes, etc.).
• Proficiency in Python and C++, with experience writing production-grade libraries, orchestration layers, and automation tools.
• Ability to work closely with multi-functional teams and translate requirements into robust systems.

Responsibilities

• Crafting, scaling, and hardening deep learning infrastructure libraries and frameworks for training on multi-thousand GPU clusters.
• Improving efficiency throughout the training stack: data loaders, distributed training, scheduling, and performance monitoring.
• Building robust training pipelines and libraries to handle massive video datasets and enable rapid experimentation.
• Collaborating with researchers, model engineers, and internal platform teams to enhance efficiency, minimize stalls, and improve training availability.
• Owning core infrastructure components such as orchestration libraries, distributed training frameworks, and fault-resilient training systems.

Skills

Python

C++

PyTorch

Distributed systems

NCCL / FSDP

Kubernetes

Education

BS/MS/PhD in CS or related field

Tools

RoCE

Lustre

Slurm

Job description

NVIDIA is seeking a Senior Deep Learning Systems Engineer to advance its Autonomous Vehicles project. You will build and scale training libraries and infrastructure to enable end-to-end autonomous driving models on thousands of GPUs, collaborating with research and platform teams to boost efficiency and availability.

Ideal candidates have 12+ years with deep learning systems, strong Python/C++, PyTorch expertise, and a solid systems background in data centers, networking, and orchestration