We are seeking researchers to investigate acceleration of DL training and inference with GPU systems. This is an applied research position, where the goal is to investigate various techniques and to transfer the most promising ones to SW and HW products. Work entails investigations, prototyping, publication, and collaboration with researchers, DL practitioners, and engineering teams within and outside NVIDIA.
What you'll be doing:
- Numerics: research of low-bit number representations and their effect on neural network inference and training accuracy. This includes requirements by the existing state of art neural networks, as well as co-design of future neural network architectures and optimizers.
- System performance: research various parallelization approaches for large neural networks (both training and inference), communication patterns and their performance limiters on large GPU systems, communication and computation overlap.
Want more jobs like this?
Get Science and Engineering jobs in Santa Clara, CA delivered to your inbox every week.
What we need to see:
- MS or PhD degree in computer science, computer engineering or a related field. Equivalent experience in some of the areas listed below can substitute for an advanced degree.
- At least 3+ years of relevant industry experience.
- Familiarity with state of art neural network architectures, optimizers.
- Experience with modern DL training frameworks and/or inference engines.
- Fluency in Python, C++, or ideally both.
- For numerics focused candidates:
- Experience with training neural networks (LLMs and multi-modal models are of particular interest), exploring model architectures and optimizers.
- Experience with quantization of neural networks, numerical analysis, number representations and computer arithmetic.
- For the systems focused candidates:
- Experience with parallel programming and performance analysis, collective communications.
- Background in computer architecture. Experience with GPU computing and CUDA is not required but a big plus.
Intelligent machines powered by AI computers that can learn, reason and interact with people are no longer science fiction. Today, a self-driving car powered by AI can meander through a country road at night and find its way. An AI-powered robot can learn motor skills through trial and error. This is truly an extraordinary time. The era of AI has begun.
NVIDIA is widely considered to be one of the technology world's most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us. If you're creative, autonomous and love a challenge, we want to hear from you!
The base salary range is 160,000 USD - 345,000 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.
You will also be eligible for equity and benefits. NVIDIA accepts applications on an ongoing basis.
NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.