The Software Engineer Intern role at NVIDIA focuses on developing innovative technologies in deep learning inference and training systems software. As part of the DL Kernel Library team, interns will work on cutting-edge AI systems software, including cuDNN and FlashInfer. The position involves designing and building efficient kernel implementations and code generators for NVIDIA's hardware architecture. Interns will collaborate with other engineers and contribute to open source communities, making a significant impact in the AI field.

• BS, MS, or PhD degree in Computer Science, Electrical Engineering, or a related field or equivalent experience.
• Strong experience in developing or using deep learning frameworks such as PyTorch, JAX, TensorFlow, or ONNX.
• Strong Python and C/C++ programming skills.

• Innovate and develop new AI systems technologies for efficient inference and training.
• Design, implement, and optimize kernels for high impact AI workloads.
• Design and implement extensible abstractions for serving engines and training frameworks.
• Build efficient just-in-time domain specific compilers and runtimes.
• Collaborate closely with other engineers at NVIDIA across deep learning frameworks, libraries, kernels, and GPU architecture teams.
• Contribute to open source communities across FlashInfer, cuDNN, vLLM, SGLang, PyTorch, JAX, and XLA.

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