Senior Software Engineer: Numerical and High Performance Computing
OpenTeamsSalary not specified
Full Time
Senior (5 to 8 years)
Not SpecifiedCompensation
Senior (5 to 8 years)Experience Level
Full TimeJob Type
UnknownVisa
Technology, Artificial Intelligence, High Performance Computing, BiotechnologyIndustries
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++, 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, self-motivated, and excellent teammate
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
Skills
Key technologies and capabilities for this role
Questions & Answers
Common questions about this position
What compensation and benefits does NVIDIA offer for this role?
NVIDIA offers highly competitive salaries and a comprehensive benefits package.
Is this position remote or does it require working in an office?
This information is not specified in the job description.
What skills and experience are required for the Senior HPC Performance Engineer role?
Candidates need an advanced degree in a quantitative field, 8+ years of relevant experience, deep understanding of parallel programming in C++, Python (ideally CUDA and OAI Triton), fluency in ML frameworks like PyTorch, TensorFlow, JAX, and experience with HPC solutions.
What is the company culture like at NVIDIA for this team?
The team consists of scientists and engineers passionate about building next-generation scientific ML frameworks, with forward-thinking, resourceful, and talented people; they value diversity, strong communication, organization, self-motivation, and excellent teamwork.
What makes a candidate stand out for this Senior HPC Performance Engineer position?
Stand out with contributions to major scientific AI for Science codebases and familiarity with pioneering language and geometric models in AI for Science applications in biology, chemistry, or material science, along with a track record in performance engineering, technical leadership, and ability to learn and teach others.
NVIDIA
Designs GPUs and AI computing solutions
About NVIDIA
NVIDIA designs and manufactures graphics processing units (GPUs) and system on a chip units (SoCs) for various markets, including gaming, professional visualization, data centers, and automotive. Their products include GPUs tailored for gaming and professional use, as well as platforms for artificial intelligence (AI) and high-performance computing (HPC) that cater to developers, data scientists, and IT administrators. NVIDIA generates revenue through the sale of hardware, software solutions, and cloud-based services, such as NVIDIA CloudXR and NGC, which enhance experiences in AI, machine learning, and computer vision. What sets NVIDIA apart from competitors is its strong focus on research and development, allowing it to maintain a leadership position in a competitive market. The company's goal is to drive innovation and provide advanced solutions that meet the needs of a diverse clientele, including gamers, researchers, and enterprises.
Santa Clara, CaliforniaHeadquarters
1993Year Founded
$19.5MTotal Funding
IPOCompany Stage
Automotive & Transportation, Enterprise Software, AI & Machine Learning, GamingIndustries
10,001+Employees