27 applicants

Company

NVIDIA · 1 day ago

Senior HPC Performance Engineer - AI for Science at Scale

North Carolina, United States

Full-time

Remote

Senior Level, Lead/Staff

$180K/yr - $339K/yr

8+ years exp

Maximize your interview chances

Artificial Intelligence (AI)GPU

Growth Opportunities

H1B Sponsor Likely

Insider Connection @NVIDIA

Discover valuable connections within the company who might provide insights and potential referrals.
Get 3x more responses when you reach out via email instead of LinkedIn.

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

Find out how your skills align with this job's requirements. If anything seems off, you can easily click on the tags to select or unselect skills to reflect your actual expertise.

C++PythonCUDA programmingMachine learning frameworksHPC solutionsPerformance engineeringSoftware designData processing toolsAlgorithm developmentDigital biologyComputational chemistryMaterial science applications

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

Consistent 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

Fluent 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

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 and benefits

Company

NVIDIA

Glassdoor

4.6

NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.

Founded in 1993

Santa Clara, California, USA

10001+ employees

https://www.nvidia.com

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

2023 (735)

2022 (892)

2021 (696)

2020 (534)