NVIDIA is seeking a motivated Deep Learning engineer to integrate advanced CUDA features and Distributed Runtime technologies into AI stacks such as PyTorch and JAX. The role involves collaborating with a dynamic team that focuses on enhancing performance and programmability for AI applications. The engineer will work on multi-GPU demands, contributing to the development of innovative technologies that align with NVIDIA's vision. This position offers an opportunity to advance the state of the art in deep learning and high-performance computing.

• BS, MS, or PhD degree in Computer Science, Computer Engineering, Electrical Engineering, or related field.
• 8+ years of relevant industry experience or equivalent academic experience after completed degree.
• Development experience with Deep Learning Frameworks such as PyTorch and JAX.
• Experience with Inference Engines such as TRT-LLM, vLLM, and SGLang.
• Solid grasp of AI models, parallelisms, and/or compiler technologies.
• Experience conducting performance benchmarking on AI clusters.
• Good understanding of computer system architecture and operating systems principles.
• Adaptability and passion to learn new frameworks and tools.

• Integrate new CUDA features and Runtime abstractions in AI frameworks from PoC to performance analysis to production.
• Perform deep analysis of AI workloads and frameworks to identify requirements and opportunities for innovation.
• Own and drive improvements in the AI Compiler-Runtime interface for multi-GPU multi-node solutions.
• Design fault-tolerant and elastic solutions for large-scale AI workloads.
• Influence the roadmap of core CUDA to facilitate building next-gen deep learning frameworks.
• Collaborate closely with AI researchers, HW and SW architects, and CUDA driver experts.
• Develop exploratory tools and runtime systems to profile and accelerate new paradigms in deep learning.
• Write clean, effective, and maintainable code for various applications.

• 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.