Differentiable Simulation Research Intern - Summer 2027
$0k - $0k • Remote • US, CA, Santa Clara
Posted 7d ago
Job Location
US, CA, Santa Clara
Tech Stack
Remote Work Policy
Fully remote
Categories
Machine Learning Engineer
About the job
NVIDIA is seeking interns to research scalable differentiable algorithms, including adjoint and automatic differentiation methods. This role will involve connecting simulations with experimental data for calibration and data assimilation using AI/ML techniques. The work will be evaluated on simulation workloads and assessed for integration into NVIDIA software. This internship is ideal for individuals passionate about mathematics, code, hardware, and engineering applications.
Responsibilities
- Prototype GPU-native solvers using adjoint and automatic differentiation for engineering simulations.
- Integrate AI-based methods with simulation algorithms for parameter calibration, inverse problems, and data assimilation.
- 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 various simulation workloads.
- Identify promising methods for integration into CUDA-X libraries.
Requirements
- Currently pursuing a PhD in applied mathematics, scientific computing, computational engineering, computer science, mechanical engineering, aerospace engineering, or a related field.
- Strong foundation 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, or gradient-aware simulation approaches.
- Experience writing numerical software in Python and C++.
- Familiarity with parallel or GPU computing.
- Ability to design computational experiments, evaluate tradeoffs, and communicate technical results.
Benefits
- Internship hourly rates based on position, location, year in school, degree, and experience.
- Eligibility for Intern benefits.