Nvidia - Santa Clara, CA

posted 2 months ago

Full-time - Entry Level
Santa Clara, CA
Computer and Electronic Product Manufacturing

About the position

The Research Scientist position at NVIDIA focuses on efficient deep learning, aiming to push the boundaries of research while making a real-world impact. The role involves working within a collaborative research team that excels in publishing high-quality research in computer vision and machine learning. The successful candidate will engage in various innovative methods, including post-training model optimization and efficient architecture design, contributing to the development of impactful technologies.

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.

Requirements

  • Completing or recently completed a Ph.D. in Computer Science/Engineering, Electrical Engineering, or equivalent research experience.
  • Excellent knowledge of theory and practice of computer vision methods and deep learning.
  • Background in pruning, quantization, NAS, and efficient backbones is expected.
  • 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) is a plus.
  • Hands-on experience with large-scale model training including data preparation and model parallelization (tensor and pipeline) is required.
  • Outstanding research track record.
  • Excellent communication skills.

Nice-to-haves

  • Experience with large language models and large vision-language models.
  • Knowledge of CUDA programming.

Benefits

  • Equity options
  • Comprehensive health benefits
  • Diversity and inclusion programs
  • Flexible work hours
  • Professional development opportunities
© 2024 Teal Labs, Inc
Privacy PolicyTerms of Service