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
CUDA
C++
Python
PyTorch
TensorFlow
JAX
Warp
Triton
HPC
GPU programming
parallel programming
distributed learning
ML frameworks
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