Senior Physics-Machine Learning Engineer - CAE
Core
Developing NVIDIA PhysicsNemo, an AI framework for constructing digital twins and machine learning simulation surrogates to solve real-world science and engineering problems.
Role type
Senior IC physics-machine learning engineer
Builds
Physics-AI framework (NVIDIA PhysicsNemo) for academic and industrial partners
Domain
Scientific computing, deep learning, multi-physics simulations (CFD, structural, thermal, etc.)
Deliverable
production ML models
Required skills
Python programming, deep learning frameworks (PyTorch, Tensorflow, JAX), DNN architectures (graph networks, diffusion models, reinforcement learning), multi-physics simulation experience, scientific visualization
Preferred skills
Multi-node systems programming (data/model parallel), CUDA, nonlinear simulation tools, published papers in AI for scientific computing
Technologies
PyTorch, Tensorflow, JAX, CUDA
Responsibilities
Develop and enhance deep learning technologies for simulations, validate product with industrial applications, implement new deep learning ideas
Seniority
Senior, hands-on IC