Taro Logo

Senior Performance Software Engineer, Deep Learning Libraries

NVIDIA is the world leader in accelerated computing, pioneering solutions for AI and digital twins.
Santa Clara, CA, USAAustin, TX, USAHillsboro, OR, USA
$184,000 - $425,500
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
6+ years of experience
AI

Description For Senior Performance Software Engineer, Deep Learning Libraries

NVIDIA is seeking a Senior Performance Software Engineer to join their Deep Learning Libraries team. This role focuses on developing optimized code to accelerate linear algebra and deep learning operations on NVIDIA GPUs. The position involves working with cutting-edge technologies like cuDNN, cuBLAS, and TensorRT libraries to accelerate deep learning models. The successful candidate will be instrumental in enabling breakthroughs in image classification, speech recognition, and natural language processing.

The role requires expertise in writing highly optimized compute kernels using C++ CUDA, with a focus on core deep learning operations like matrix multiplies, convolutions, and normalizations. You'll be working closely with various teams across NVIDIA, including the CUDA compiler team, deep learning performance teams, and hardware architecture teams. This position is particularly exciting as it deals with code lower in the deep learning software stack, right down to the GPU hardware level.

NVIDIA offers a competitive compensation package with a base salary range of $184,000 - $425,500 USD, plus equity and benefits. The company is known for being one of the technology world's most desirable employers, offering opportunities to work on groundbreaking projects in AI and accelerated computing. They're committed to fostering a diverse work environment and are proud to be an equal opportunity employer.

The ideal candidate will have at least 6 years of relevant industry experience, strong C++ programming skills, and experience with parallel programming. Additional expertise in CUDA/OpenCL GPU programming, numerical methods, linear algebra, or tools like LLVM and TensorFlow MLIR would be particularly valuable. This is an opportunity to join a team that's building the fundamental software powering the AI revolution worldwide.

Last updated a day ago

Responsibilities For Senior Performance Software Engineer, Deep Learning Libraries

  • Writing highly tuned compute kernels in C++ CUDA for core deep learning operations
  • Following software engineering best practices including regression testing and CI/CD flows
  • Collaborating with CUDA compiler team on generating optimal assembly code
  • Working with deep learning training and inference performance teams
  • Collaborating with hardware and architecture teams on programming models

Requirements For Senior Performance Software Engineer, Deep Learning Libraries

  • Masters or PhD degree or equivalent experience in Computer Science, Computer Engineering, Applied Math, or related field
  • 6+ years of relevant industry experience
  • Strong C++ programming and software design skills
  • Experience with performance-oriented parallel programming
  • Solid understanding of computer architecture and assembly programming

Benefits For Senior Performance Software Engineer, Deep Learning Libraries

Equity
  • Equity
  • Benefits package

Interested in this job?

Jobs Related To NVIDIA Senior Performance Software Engineer, Deep Learning Libraries

Senior Software Engineer - Robotics and AI

Senior Software Engineer position at NVIDIA focusing on robotics and AI, developing advanced solutions for humanoid robots and embodied agents.

Senior Software Engineer, Deep Learning Inference

Senior Software Engineering role at NVIDIA focusing on developing and optimizing deep learning inference frameworks for Large Language Models (LLMs) on GPU platforms.

Senior Software Engineer, Digital Human Technology

Senior Software Engineer position at NVIDIA focusing on Digital Human Technology, combining AI research with software engineering to develop cutting-edge digital human solutions.

Senior Deep Learning Software Engineer, LLM Performance

Senior Deep Learning Software Engineer position at NVIDIA focusing on LLM performance optimization, offering competitive salary and opportunity to work with cutting-edge AI technology.

Senior Software Engineer, Robotics Foundation Models

Senior Software Engineer role at NVIDIA focusing on robotics foundation models, AI, and computer vision for humanoid robots.