NVIDIA is looking to solve the problem of efficient deep learning to create real impact on their products
Requirements
- Excellent knowledge of theory and practice of computer vision methods, as well as deep learning
- Background in pruning, quantization, NAS, efficient backbones, and so on, is a plus
- Experience with large language models and large vision-language models is required
- Excellent programming skills in Python and PyTorch; C++ and parallel programming (e.g., CUDA) is a plus
- 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
- Collaborate with other team members and teams
- Mentor interns
- Speak at conferences and events
- Work with product groups to transfer technology
- Collaborate with external researchers
Other
- Completing or recently completed a Ph.D. in Computer Science/Engineering, Electrical Engineering, etc., or have equivalent research experience
- Outstanding research track record
- Excellent communications skills
- Travel requirements not specified
- Visa requirements not specified