Senior HPC Performance Engineer, AI for Science at Scale

September 30

🏄 California – Remote

Although this job is listed in California, this company may be open to hiring remote in other states.

🌲 North Carolina – Remote

Although this job is listed in North Carolina, this company may be open to hiring remote in other states.

+3 more states

🏄 California – Remote 🌲 North Carolina – Remote 🌲 Oregon – Remote 🍂 Massachusetts – Remote 🤠 Texas – Remote

💵 $184k - $356.5k / year

⏰ Full Time

🟠 Senior

🤖 Artificial Intelligence

🦅 H1B Visa Sponsor

Apply Now

Receive Emails with Similar Jobs

NVIDIA

WebsiteLinkedInAll Job Openings

Artificial Intelligence • Gaming • Automotive

NVIDIA is a leading technology company specializing in accelerated computing and artificial intelligence. NVIDIA pioneers advancements in graphical processing units (GPUs), cloud computing, data centers, and virtual reality, with a focus on gaming, automotive, healthcare, and robotics industries. The company's innovations, such as NVIDIA Omniverse, transform traditional digital processes by enabling high-fidelity simulations and rendering tasks. Their applications span various industries, from autonomous vehicles using NVIDIA DRIVE to healthcare solutions with NVIDIA Clara, and AI-driven analytics and workflows.

10,000+ employees

Founded 1993

🤖 Artificial Intelligence

🎮 Gaming

📋 Description

• 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

🎯 Requirements

• 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++ and Python; ideally CUDA and OAI Triton 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, organized and self-motivated, excellent teammate • (Ways to stand out) Contributor to major scientific AI for Science codebase • (Ways to stand out) Familiarity with pioneering language and geometric models used in AI for Science applications in biology, chemistry, material science

🏖️ Benefits

• With highly competitive salaries and a comprehensive benefits package • You will also be eligible for equity and benefits