NVIDIA Corporation

Senior HPC Performance Engineer - AI for Science at Scale

NVIDIA Corporation$184K — $287K *
Information Technology
5 - 7 years of experience
Job Overview by Ladders

Qualifications

  • Advanced degree in Computer Science, Computational Biophysics, Computational Chemistry, Physics, Mathematics, or equivalent experience
  • 5+ years of relevant experience in performance engineering and software design
  • Proven experience launching software products with a focus on acceleration
  • Strong skills in parallel programming using C++ and Python
  • Experience with CUDA or OAI Triton programming
  • Familiarity with modern machine learning frameworks (PyTorch, JAX, Warp)
  • Experience with HPC solutions for biology or chemistry research

Responsibilities

  • Design and implement high-performance features for large-scale ML training frameworks
  • Optimize the computation performance of critical ML models with hardware and software stack improvements
  • Develop and maintain HPC software stack for atomistic modeling and generative machine learning applications
  • Collaborate with HPC, AI infrastructure, and research teams on various projects
  • Test and maintain algorithms and software modules

Benefits

  • Highly competitive salaries
  • Comprehensive benefits package
  • Eligibility for equity options
  • Work in a diverse and inclusive environment
  • Opportunity to collaborate with top talent in the industry
Full Job Description
What you'll be doing:
  • Design and implement computationally performant features for large scale, CUDA-backed ML training frameworks, using low level acceleration and scaling strategies such as kernel design, 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 atomistic modeling and generative machine learning in digital biology and beyond
  • Collaborate with multiple HPC, AI infrastructure, and research teams
  • Drive the testing and maintenance of the algorithms and software modules

What we need to see:
  • Advanced degree in a quantitative field such as Computer Science, Computational Biophysics, Computational Chemistry, Physics, Mathematics, or equivalent experience
  • 5+ years of relevant experience
  • Consistent track record in performance engineering as well as software design, building and packaging and launching software products, with a focus on acceleration
  • Deep understanding of parallel programming in C++, Python; programming experience CUDA or OAI Triton
  • Fluent in modern machine learning frameworks such as PyTorch, JAX, Warp
  • Experience with HPC solutions to research problems for biology or chemistry, including but not limited to atomistic simulations
  • 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!)

Ways to stand out from the crowd:
  • Contribution to major scientific AI for Science codebase with acceleration features such as new kernels
  • Familiarity with pioneering language and geometric models used in AI for Science applications in biology and chemistry

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184,000 USD - 287,500 USD.

You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until February 21, 2026.

This posting is for an existing vacancy.

NVIDIA uses AI tools in its recruiting processes.

About NVIDIA Corporation

Nvidia, a global leader in graphics, gaming, and AI technology, offers Nvidia careers and internship opportunities for those passionate about driving innovation in the tech industry. you'll find a company committed to growth, teamwork, and leadership in computer science and machine learning domains.

About Nvidia

A Pioneer in Technology and Innovation

Nvidia has cemented its reputation as a powerhouse in developing advanced graphics processing units (GPUs) and has significantly contributed to the gaming industry's evolution. Moreover, its foray into AI and machine learning has opened new frontiers in technology, making Nvidia a beacon of innovation and a desirable workplace for ambitious tech professionals.

Job Opportunities

Diverse Positions in a Dynamic Field

Nvidia is continuously on the lookout for talented individuals across various domains, including hardware and software engineering, product design, marketing, and sales. Employment opportunities at Nvidia are vast, catering to a wide range of expertise and career aspirations.

Employment in Hardware and Graphics

For those fascinated by the intricacies of hardware and graphics technology, Nvidia offers positions that sit at the forefront of gaming and computing advancements.

Growth in Machine Learning and AI

Nvidia's leadership in AI and machine learning has created numerous vacancies for specialists eager to contribute to groundbreaking projects.

Recruitment in Computer Science

With the constant demand for innovation, Nvidia's recruitment efforts focus on computer science experts capable of pushing the boundaries of what's possible.

Internship Program

Opening Doors to Future Innovators

Nvidia's internship program is designed to nurture the next generation of technology leaders, offering hands-on experience in a culture that celebrates creativity and teamwork.

Benefits and Culture

Interns at Nvidia enjoy a plethora of benefits, from competitive stipends to mentorship opportunities, all within an environment that values growth and learning.

Opportunities for Students

Whether you're an undergraduate, a master's student, or a Ph.D. candidate, Nvidia's internships provide a real-world glimpse into the tech industry, offering valuable experience in various technology fields.

Pathways to Full-Time Employment

Many interns have transitioned into full-time positions, marking the start of successful careers at Nvidia. The internship program is more than a stepping stone into the company; it’s an investment in the professional development of interns. The goal is to ensure that interns are well-equipped for future challenges.

Nvidia Careers: More Than Just a Job

Nvidia offers more than just a job to its employees; it provides a front-row seat on the journey into the future of technology. Nvidia stands as a pillar of innovation with its vast opportunities in hardware, graphics, gaming, machine learning, and computer science. Nvidia careers serve as a launching pad for talented workers who aim to redefine the technological landscape. Whether through full-time positions or internships, joining Nvidia means contributing to a legacy of breakthroughs and becoming part of a global community dedicated to pushing the boundaries of what's possible.
Learn more about NVIDIA Corporation
Size
22,473 employees
Market Cap
$350.4 billion
Industry
Net Income
$4.3 billion
Founded
1993
5 Year Trend
+31.3%
Revenue
$16.6 billion
NASDAQ

Similar Jobs

More Jobs at NVIDIA Corporation

More Information Technology Jobs

Find similar Senior HPC Performance Engineer - AI for Science at Scale jobs: