Senior Systems Software Engineer, TAO Deep Learning

at Nvidia

📍 Santa Clara, United States

$148,000-276,000 per year

SENIOR
✅ On-site

SCRAPED

Used Tools & Technologies

Not specified

Required Skills & Competences ?

Python @ 7 Algorithms @ 4 Machine Learning @ 4 TensorFlow @ 7 Hiring @ 4 Communication @ 7 Mentoring @ 4 PyTorch @ 7

Details

NVIDIA is hiring a Senior Systems Software Engineer, Deep Learning to join the TAO Toolkit Deep Learning Architectures team. Our team develops scalable and pioneering training, fine-tuning, and optimization algorithms for Computer Vision and Multi-Modal AI, to help advance the state of the art while improving performance.

We are seeking someone who can help advance, scale, and optimize our Deep Learning Architectures. If you have a passion for brand-new technologies and a commitment to developing scalable, optimized, and ethical AI, we invite you to join our dynamic team at NVIDIA. In this role, you will be developing state-of-the-art algorithms to train computer vision models and implementing methods to optimize these models for latency.

Responsibilities

  • Architect, analyze, develop, and prototype key deep learning algorithms and solutions as a core member of our growing software team.
  • Collaborate with diverse software, research, and hardware teams across geographies to analyze the interplay of hardware and software architectures and solve critical problems and future applications.
  • Develop algorithms (such as zero/few-shot learning, unsupervised learning) to address data scarcity and collection challenges.
  • Apply generative models (Diffusion, GANs, VAEs) and LLMs for data generation to overcome data scarcity issues.
  • Drive the design and implementation of complex AI projects, providing technical guidance and support, mentoring junior engineers.
  • Create and refine algorithms for a varied number of computer vision and multi-modal tasks.

Requirements

  • 3 years or more of working experience.
  • MS or PhD or equivalent experience in Computer Science, Computer Engineering, Electrical Engineering, or a related field with a focus on Deep Learning, Machine Learning, and Computer Vision.
  • Experience in algorithm development for AI, computer vision, or multi-modal algorithms, especially with LLMs and Multi-Modal Foundation models.
  • Experience working with and curating multi-modal datasets.
  • Experience with algorithms including zero/few-shot learning, fully-supervised, weakly-supervised, self-supervised, and unsupervised learning techniques, and domain adaptation techniques like Parameter Efficient Fine-Tuning.
  • Proficiency in working with deep learning frameworks such as TensorFlow and PyTorch, strong programming skills in Python and/or C++, and experience developing integrated AI solutions.
  • Ability to lead projects, manage timelines, and deliver results.
  • Expert analytical and problem-solving skills with a focus on practical and scalable AI solutions.
  • Strong communication skills and ability to work in a collaborative environment.

Benefits

NVIDIA offers a base salary range of 148,000 - 276,000 USD. You will also be eligible for equity and benefits. NVIDIA is committed to fostering a diverse work environment and is proud to be an equal opportunity employer.