
Senior Deep Learning Frameworks CUDA Software Engineer
NVIDIA is hiring a Senior Deep Learning Frameworks CUDA Software Engineer to add advanced CUDA features and runtime abstractions into AI stacks (PyTorch, JAX, TRT-LLM, vLLM, SGLang). Responsibilities include performance analysis, integrating features from PoC to production, designing fault-tolerant elastic multi-GPU/multi-node solutions, and collaborating across HW/SW/research teams. Requirements: BS/MS/PhD or equivalent, 8+ years industry experience, experience with deep learning frameworks, C++/Python/CUDA development, profiling and benchmarking, and distributed/HPC communication (NCCL/MPI/UCX).