← All roles

Senior Deep Learning Framework Communications Engineer

NVIDIA· Engineer, Sys SW
Apply on company site
Hybrid (Remote / Local)$184k - $287.5kfull-timePosted 7mo ago

About the role

Senior role integrating and optimizing communication libraries and AI frameworks to scale multi-GPU deep learning workloads, spanning PoC to production, across time zones. Focus on NCCL/NVSHMEM, AI toolkits, and large-scale AI workloads.

Responsibilities

  • Integrate new communication libraries features in AI frameworks: from PoC to performance analysis to production
  • Perform deep analysis of AI workloads and frameworks to identify multi-GPU communication requirements and opportunities. Collaborate hands-on with teams working on the latest AI models.
  • Improve AI compilers to hide communications or perform automatic fusion.
  • Conduct in-depth AI workload performance characterization on multi-GPU clusters.
  • Design fault-tolerant and elastic solutions for large-scale or dynamic AI workloads.
  • Authorize custom communication or fused compute-communication kernels to showcase ultimate performance on NV platforms.
  • Influence the roadmap of communication libraries - NCCL & NVSHMEM.
  • Collaborate with a very dynamic team across multiple time zones.

Qualifications

  • B.S, M.S. or PHD in Computer Science, or related field (or equivalent experience) with 8+ software engineering and HPC/AI experience
  • Development or integration experience with Deep Learning Frameworks such PyTorch, JAX, and Inference Engines such as TRT-LLM, vLLM, SGLang
  • Rapid prototyping and development with Python, C++, CUDA or related DSLs (Triton, cuTe)
  • Solid grasp of AI models, parallelisms, and/or compiler technologies (e.g. torch.compile)
  • Experience conducting performance benchmarking on AI clusters. Familiarity with at least one performance profiler toolchain (PyTorch profiler, NVIDIA Nsight Systems)
  • Understanding of HPC/AI communication concepts (1-sided v 2-sided communication, elasticity, resiliency, topology discovery, etc)
  • Adaptability and passion to learn new areas and tools
  • Flexibility to work and communicate effectively across different teams and timezones

Skills

Required:Deep Learning Framework integrationProgrammingAI models, parallelisms, and/or compiler technologiesPerformance benchmarkingHPC/AI communication concepts

Benefits

  • equity
  • benefits (mentioned on benefits page)
Last updated Oct 23, 2026. Interested? Apply on the company site.

About NVIDIA

World Leader in AI Computing

NVIDIA invents the GPU and drives advances in AI, HPC, gaming, creative design, autonomous vehicles, and robotics.

AIautonomous vehiclesgaminggpugraphics cardhigh performance computingliquid-ai-recruiting-targetllm
Company profileWebsite

More open roles at NVIDIA