N

Senior Performance Software Engineer, Deep Learning Libraries

NVIDIA

Hayward, CA
4 days ago
$184,000 - $425,500

Job Description

Join NVIDIA as a Senior Performance Software Engineer for Deep Learning Libraries, where you'll develop optimized code to accelerate deep learning operations on NVIDIA GPUs. Collaborate with cross-functional teams and contribute to innovations in AI by enhancing GPU performance.

Key Responsibilities

  • 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 various teams at NVIDIA for optimization and programming model development

Required Qualifications

  • 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, including debugging, performance analysis, and test design
  • Experience with performance-oriented parallel programming
  • Solid understanding of computer architecture and some experience with assembly programming

Preferred Qualifications

  • Tuning BLAS or deep learning library kernel code
  • CUDA/OpenCL GPU programming
  • Numerical methods and linear algebra
  • LLVM, TVM tensor expressions, or TensorFlow MLIR

Benefits & Perks

EquityComprehensive benefits package