NVIDIA · 17 hours ago
Senior Machine Learning Performance Engineer - Physics
Santa Clara, CA
Full-time
Onsite
Senior Level
$152K/yr - $242K/yr
5+ years expNVIDIA is a leading AI company known for its deep learning and HPC platforms that impact various fields. They are seeking a Senior Machine Learning Performance Engineer to develop and optimize a machine learning framework for real-world science and engineering challenges.
AI InfrastructureArtificial Intelligence (AI)Consumer ElectronicsFoundational AIGPUHardwareSoftwareVirtual Reality
Responsibilities
Work with some of the brightest minds in a leading AI company to develop a leading machine learning framework, NVIDIA PhysicsNeMo, for our academic and industrial partners to construct digital twins and machine learning simulation surrogates for real world science and engineering problems
Work with internal project teams to validate applications built using the framework on Nvidia’s products
Characterize and optimize scientific AI workloads on the latest NVIDIA platforms
Stay up to date with the latest research and innovations in deep learning techniques, implement and experiment with new ideas to develop and enhance NVIDIA's deep learning technologies with focus on simulations
Qualification
Required
BS or MS degree (PhD preferred) in computer science, mathematics, computational science/engineering, or related technical field or equivalent experience
5+ yrs of relevant experience. Strong Python programming skills
Familiarity with containers, numeric libraries, modular software design, distributed computing, and/or high-performance storage systems
Good knowledge of state-of-the-art DNN architectures and machine learning techniques and algorithms (graph networks, diffusion models, reinforcement learning etc.) with experience in developing or using major deep learning frameworks (PyTorch, Tensorflow, JAX etc.)
Experience with solving and using machine learning for real world problems involving scientific/engineering simulations (domains/applications - industrial, life sciences, high energy physics, earth sciences – seismic, weather & climate modeling; physics types - CFD, structural, electromagnetics, optics, acoustics etc.)
Experience with scientific visualization is a big plus. Experience with performance measurement tools, such as NVIDIA Nsight Systems
Strong analytical skills with bias for action. Good time-management and organization skills to thrive in a fast paced, dynamic environment
Solid written and oral communications skills
Good teamwork and interpersonal skills
Preferred
Work with multi-node systems with data-parallel and model parallel programming experience. Experience with CUDA, or python kernel languages such as Triton or NVIDIA Warp
Usage of nonlinear simulation tools and techniques, usage of major simulation codes (opensource and/or commercial). Experience with High Performance Computing for Scientific AI, or end-to-end training and inference optimization
Published papers in the field of AI in scientific computing
Benefits
Equity
Benefits
Company
NVIDIA
NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.
10001+ employees
H1B Sponsorship
NVIDIA has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1877)
2024 (1355)
2023 (976)
2022 (835)
2021 (601)
2020 (529)
Funding
Current Stage
Public CompanyTotal Funding
$4.09BKey Investors
ARPA-EARK Investment ManagementSoftBank Vision Fund
2023-05-09Grant· $5M
2022-08-09Post Ipo Equity· $65M
2021-02-18Post Ipo Equity
Leadership Team
Jensen Huang
Founder and CEO
Michael Kagan
Chief Technology Officer
Recent News
2026-02-06
The Real Deal
2026-02-06
2026-02-06
Company data provided by crunchbase