5 days ago
Base Salary
$184k - $357k/yr
Responsibilities
- Design and build PyTorch components that run efficiently on systems with thousands to hundreds of thousands of GPUs.
- Collaborate with NVIDIA hardware and software teams to improve GPU performance in PyTorch.
- Design, build, and support production AI solutions for enterprise customers and partners.
- Work with internal applied researchers to improve AI tools.
- Collaborate with multiple NVIDIA teams and the international PyTorch community.
Requirements
- Bachelor’s degree in computer science or engineering, or equivalent experience.
- At least 3 years of professional experience in deep learning.
- Proficiency with C++ programming.
- Strong understanding of systems software and interfaces.
- Demonstrated experience with threading and distributed parallel programming.
- Demonstrated background developing large software projects.
- Strong verbal and written communication skills.
- Preferred qualifications include open-source contributions, familiarity with deep learning compilers and modeling trends, CUDA and Python experience, and experience working with multidisciplinary teams.
Benefits
- Hybrid work arrangement.
- Equity and comprehensive benefits package.
- The posting includes Level 4 and Level 5 base salary ranges, with compensation determined by location, experience, and comparable employee pay.
Categories
About Nvidia
Nvidia designs and sells GPUs and accelerated computing platforms for data centers, AI/ML, graphics, gaming, and automotive, monetizing through hardware, software platforms (CUDA, AI frameworks), and systems like DGX and networking. Customers include cloud providers, enterprises, researchers, and OEMs. Founded in 1993 and headquartered in Santa Clara, it is a public company traded on NASDAQ under NVDA.