NVIDIA is looking to solve the problem of efficient deep learning to create real impact on their products through research and development of novel methods.
Requirements
- Excellent knowledge of theory and practice of computer vision methods, as well as deep learning.
- A background in pruning, quantization, NAS, efficient backbones is required.
- Experience with large language models and large vision-language models is a plus.
- Excellent programming skills in Python and PyTorch; C++ and parallel programming (e.g., CUDA).
- Hands-on experience with large-scale model training including data preparation and model parallelization (tensor and pipeline) is required.
Responsibilities
- Research, design and implement novel methods for efficient deep learning.
- Publish original research. Speak at conferences and events.
- Collaborate on research with internal team members, internal teams as well as external researchers. Mentor interns.
- Work with product groups to transfer technology.
Other
- A Ph.D. in Computer Science/Engineering, Electrical Engineering, etc., or equivalent experience in industrial research labs.
- 3+ years or relevant post-graduate research experience
- An outstanding research track record and strong communications skills.
- Travel requirements not mentioned
- Visa requirements not mentioned