# Senior Deep Learning Framework Communications Engineer

Hiring organization: [NVIDIA](https://career.thegoodapps.co/organizations/nvidia)

Canonical page: https://career.thegoodapps.co/jobs/b7761981-108e-47bb-a24a-6cf3d3923302

Listed on NVIDIA's own careers site. Applications go to them directly.

- Seniority: Senior
- Salary: 152000 – 287500 USD per year

## Summary

NVIDIA seeks a deep learning engineer to integrate advanced communication technologies into AI frameworks like PyTorch and JAX, optimizing multi-GPU performance for training and inference workloads. The role involves hands-on development with communication libraries, compiler improvements, and performance analysis across large-scale distributed systems.

_Our summary, not NVIDIA's wording._

## Skills named

C++, CUDA, JAX, Python, PyTorch

## Required

- Bachelor's degree or higher in Computer Science or related field, or equivalent experience
- 5+ years of software engineering and HPC/AI experience
- Development or integration with deep learning frameworks like PyTorch or JAX
- Experience with inference engines such as TRT-LLM, vLLM, or SGLang
- Proficiency in Python, C++, CUDA, or related domain-specific languages
- Understanding of AI models, parallelism, and compiler technologies
- Performance benchmarking experience on AI clusters with profiling tools
- Knowledge of HPC/AI communication concepts including 1-sided/2-sided communication, elasticity, and resiliency

## Nice to have

- Experience with parallel programming runtimes like NCCL, NVSHMEM, or MPI
- Strong understanding of computer system architecture and OS principles
- Expertise in training, distributed inference, MoE, or reinforcement learning
- Kernel authoring experience with CUDA, Triton, or cuTe
- Experience with compute-communication overlap in distributed runtimes
- AI compiler pattern matching and lowering expertise
- Understanding of memory hierarchy, consistency models, and tensor layouts

Apply on NVIDIA's site: https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/US-CA-Santa-Clara/Senior-Deep-Learning-Framework-Communications-Engineer_JR2011908