Senior AI-Native Numerical Methods Scientist — Solver ML

Nvidia Corporation in; Santa Clara (CA); On-site; USD 224,000 - 357,000; Full time

Benefits offered by this job

Equity

Job summary

NVIDIA is seeking an Senior Applied Research Scientist to bridge machine learning with numerical algorithms, accelerating solvers on modern GPUs and shaping AI-native numerical methods across large-scale simulations.

The role requires a PhD or equivalent, 5+ years of experience, and expertise in ML, scientific computing, and differentiable programming. Responsibilities include developing AI-guided multigrid, differentiable solvers, and solver-in-the-loop pipelines.

Qualifications

• PhD or equivalent experience in computer science, machine learning, scientific computing, applied mathematics, computational engineering, physics, or a related field.
• 5 years of relevant work/research experience.
• Background in machine learning and scientific computing, with evidence of connecting ML methods to numerical algorithms.
• Experience with PyTorch, JAX, or comparable deep learning frameworks, plus Python and GPU computing.
• Working knowledge of PDEs, sparse linear algebra, iterative solvers, preconditioning, finite element / finite volume methods, optimization, or differentiable programming.
• Research record in scientific machine learning, numerical linear algebra, or differentiable simulation integrated with numerical solvers.
• Communication skills that support collaboration across AI research, numerical methods, product, and production software teams.

Responsibilities

• Research AI-assisted numerical methods that improve convergence, stability, accuracy, robustness, and wall-clock performance for large-scale scientific, engineering, and industrial simulations.
• Invent learned coarse spaces, learned preconditioners, AI-guided multigrid methods, sequence-aware solver strategies, solver-control policies, differentiable solver components, and hybrid numerical / ML algorithms.
• Build solver-in-the-loop pipelines using residual histories, discretized operators, meshes, geometry, simulation outputs, performance counters, and physics constraints.
• Define evaluation methods that measure numerical impact, including convergence rate, failure rate, conservation, memory footprint, correctness, and end-to-end speedup.
• Collaborate with teams across numerical methods, CUDA-X, Warp, PhysicsNeMo, NVIDIA Research, CAE, EDA, semiconductor, electronics, thermal-fluid, electromagnetics, and digital twin workflows.
• Help define NVIDIA's applied research agenda for AI-native numerical methods and solver intelligence.

Skills

Machine learning

Scientific computing

Numerical methods

GPU computing

Communication skills

Education

PhD or equivalent experience

Tools

PyTorch

JAX

Python

Job description

NVIDIA is seeking an Senior Applied Research Scientist to bridge machine learning with numerical algorithms, accelerating solvers on modern GPUs and shaping AI-native numerical methods across large-scale simulations.

The role requires a PhD or equivalent, 5+ years of experience, and expertise in ML, scientific computing, and differentiable programming. Responsibilities include developing AI-guided multigrid, differentiable solvers, and solver-in-the-loop pipelines.