NVIDIA · 3 hours ago
Senior HPC Performance Engineer - AI for Science at Scale
North Carolina, United States
Full-time
Onsite
Senior Level, Lead/Staff
$184K/yr - $288K/yr
8+ years expNVIDIA is a leading platform for AI-powered applications, seeking a Senior HPC Performance Engineer to enhance scientific machine learning frameworks. The role involves designing and implementing features for CUDA-backed ML training frameworks and optimizing performance for ML models across various domains.
AI InfrastructureArtificial Intelligence (AI)Consumer ElectronicsFoundational AIGPUHardwareSoftwareVirtual Reality
Responsibilities
Design and implement computationally performant features for large scale, CUDA-backed ML training frameworks, using low level acceleration and scaling strategies such as GPU porting, data structure innovations, distributed learning technologies
Optimize computational performance of wide range of business-critical ML models via accelerated hardware and software stack, as well as algorithmic improvements
Develop and maintain HPC software stack for generative machine learning models in digital biology and beyond
Collaborate with multiple HPC, AI infrastructure, and research teams
Develop tools to assist data processing, data quality control, algorithm development, and algorithm testing
Drive the testing and maintenance of the algorithms and software modules
Qualification
Required
Advanced degree in a quantitative field such as Computer Science, Computational Biophysics, Computational Chemistry, Physics, Mathematics, or equivalent experience
8+ years of relevant experience
Proven track record in performance engineering as well as software design, building and packaging and launching software products
Deep understanding of parallel programming in C++, Python; ideally CUDA programming experience
Proficient in modern machine learning frameworks such as PyTorch, TensorFlow, JAX, Warp
Experience with HPC solutions to research problems, ideally for biology, chemistry or material science applications
Recognized for technical leadership contributions, capable of self-direction, and ability to learn from and teach others
You should display strong communication skills, be organized and self-motivated, and play well with others (be an excellent teammate!)
Preferred
Contributor to major scientific AI for Science codebase
Familiarity with pioneering language and geometric models used in AI for Science applications in biology, chemistry, material science
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
Straits Times
2026-02-05
Company data provided by crunchbase