Tech Stack
Tag name is followed by "@" symbol and proficiency level value.
About proficiency levels:
- 1-2 — basic awareness. Minimal hands-on experience, and a rudimentary understanding of the technology's purpose;
- 3-6 — daily use. Comfortable and regular usage, capable of handling common tasks and challenges related to the technology;
- 7-9 — you are an expert, you can teach others, you know all the pitfalls and tricks;
- 10 — exceptional knowledge, comprehensive understanding, and adeptness in all aspects of the technology, including advanced problem-solving. Think twice before claiming or demanding such level.
AI
Algorithms @ 6
CUDA @ 3
GPU @ 2
JAX @ 3
Machine Learning
Mathematics @ 3
Performance Analysis @ 3
Profiling @ 3
PyTorch @ 3
Python @ 2
- 1-2 — basic awareness. Minimal hands-on experience, and a rudimentary understanding of the technology's purpose;
- 3-6 — daily use. Comfortable and regular usage, capable of handling common tasks and challenges related to the technology;
- 7-9 — you are an expert, you can teach others, you know all the pitfalls and tricks;
- 10 — exceptional knowledge, comprehensive understanding, and adeptness in all aspects of the technology, including advanced problem-solving. Think twice before claiming or demanding such level.
Details
NVIDIA is seeking a research intern to work on scalable differentiable algorithms, including adjoint and automatic differentiation methods. The research will connect simulations with experimental data for calibration and data assimilation using AI/ML methods. Approaches will be evaluated on simulation workloads and assessed for integration into NVIDIA software used by researchers, engineers, and partner developers.
Responsibilities
- Prototype GPU-native solvers with adjoint and automatic differentiation techniques for engineering simulations.
- Integrate AI-based methods with simulation algorithms for parameter calibration, inverse problems, and data assimilation using experimental or simulation data.
- Co-design numerical algorithms and differentiation strategies for GPU parallelism, memory hierarchy, mixed precision, reduced synchronization, and distributed multi-GPU execution.
- Evaluate numerical accuracy, convergence, gradient quality, performance, and scalability on workloads in CFD, mechanics, thermal systems, and multiphysics.
- Identify promising methods and help move them toward reusable capabilities in CUDA-X libraries.
Requirements
- Pursuing a PhD in applied mathematics, scientific computing, computational engineering, computer science, mechanical engineering, aerospace engineering, or an equivalent program area.
- Strong foundations in algorithms related to PDEs, numerical linear algebra, iterative solvers, optimization, or scientific computing.
- Research experience with adjoint methods, automatic differentiation, inverse problems, data assimilation, gradient-aware simulation approaches, or a related area.
- Experience writing numerical software in Python and C++, with familiarity with parallel or GPU computing.
- Ability to design computational experiments, evaluate numerical and performance tradeoffs, and communicate technical results clearly.
Preferred Qualifications
- Experience developing, scaling, or analyzing GPU solvers, including multi-GPU or distributed execution.
- Hands-on CUDA experience and use of profiling, benchmarking, or performance analysis to improve numerical algorithms.
- Experience calibrating computational models or assimilating experimental data in scientific or industrial applications.
- Experience with NVIDIA Warp, CUDA-X libraries, PyTorch, JAX, PETSc, Trilinos, MFEM, libCEED, OpenFOAM, or related computational science frameworks.
- Publications, open-source contributions, or deployed research software in numerical methods, scientific computing, differentiable simulation, or computational engineering.
Benefits
Interns are eligible for NVIDIA intern benefits.
Applications will be accepted at least until October 16, 2026. This posting is for an existing vacancy. NVIDIA uses AI tools in its recruiting processes and is an equal opportunity employer.
More jobs at Nvidia
Technical Program Manager, Infrastructure Capacity and Deployment
Nvidia · Santa Clara, United States
USD 168,000-322,000 per year
Senior Staff Software Engineer — Agentic Workflows
Nvidia · Santa Clara, United States
USD 168,000-322,000 per year
Senior AI Technical Instructor
Nvidia · Santa Clara, United States
USD 200,000-322,000 per year
PhD Research Intern, Security and Privacy - 2027
Nvidia · Santa Clara, United States
USD 38-94 per hour
Senior Software Engineer, Agentic Engineering
Nvidia · Santa Clara, United States
USD 224,000-356,500 per year
Similar jobs
Senior Deep Learning Software Engineer, Inference and Model Optimization
Nvidia · Santa Clara, United States
USD 184,000-356,500 per year
PhD Research Intern, Quantum Simulation and AI - 2027
Nvidia · Santa Clara, United States
USD 38-94 per hour
Senior Software Engineer, CUDA Deep Learning Systems
Nvidia · Santa Clara, United States
USD 184,000-356,500 per year
Senior Deep Learning Systems Architect
Nvidia · Santa Clara, United States
USD 224,000-356,500 per year
Senior Software Engineer — cuEquivariance
Nvidia · Santa Clara, United States
USD 184,000-356,500 per year
Sparse Linear Algebra Libraries Intern - Summer 2027
Nvidia · Santa Clara, United States
USD 20-71 per hour
Deep Learning Software Engineer, TensorRT Performance - New College Grad 2027
Nvidia · Santa Clara, United States
USD 124,000-241,500 per year
Senior System Software Engineer, ML and Vector Search
Nvidia · Santa Clara, United States
USD 184,000-287,500 per year