Senior Performance Software Engineer Deep Learning Libraries
We are now looking for a Senior Performance Software Engineer for Deep Learning Libraries! Do you enjoy tuning parallel algorithms and analyzing their performance? If so, we want to hear from you! As a deep learning library performance software engineer, you will be developing optimized code to accelerate linear algebra and deep learning operations on NVIDIA GPUs. The team delivers
- performance code to NVIDIA’s cu
DNN, cu
BLAS, and Tensor
RTlibraries to accelerate deep learning models. The team is proud to play an integral part in enabling the breakthroughs in domains such as image classification, speech recognition, and natural language processing. Join the team that is building the underlying software used across the world to power the revolution in artificial intelligence! We’re always striving for peak GPU efficiency on current and
- generation GPUs. To get a sense of the code we write, check out our CUTLASS
- source project showcasing performant matrix multiply on NVIDIA’s Tensor Cores with CUDA. This specific position primarily deals with code lower in the deep learning software stack, right down to the GPU HW.
What you'll be doing:
- Writing highly tuned compute kernels, mostly in C++ CUDA, to perform core deep learning operations (e. g. matrix multiplies, convolutions, normalizations)
- Following general software engineering best practices including support for regression testing and CI/CD flows
- Collaborating with teams across NVIDIA:
- CUDA compiler team on generating optimal assembly code
- Deep learning training and inference performance teams on which layers require optimization
- Hardware and architecture teams on the programming model for new deep learning hardware features
What we need to see:
- Ph
D degree or equivalent experience in Computer Science, Computer Engineering, Applied Math, or related field or a Bachelors or Masters degree plus 4-6 years of equivalent relevant industry experience. - Demonstrated strong C++ programming and software design skills, including debugging, performance analysis, and test design.
- Experience with
- oriented parallel programming, even if it’s not on GPUs (e. g. with Open
MP or pthreads) - Solid understanding of computer architecture and some experience with assembly programming
Ways to stand out from the crowd:
- Tuning BLAS or deep learning library kernel code
- CUDA/Open
CL GPU programming - Numerical methods and linear algebra
- LLVM, TVM tensor expressions, or Tensor
Flow MLIR
NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most
- thinking and hard working people in the world working for us. If you're creative, autonomous, and love a challenge, consider joining our Deep Learning Library team and help us build the
- time,
- effective computing platform driving our success in this exciting and quickly growing field.
Fii primul, care se va înregistra la oferta de muncă respectivă!
-
De ce să cauți de muncă pe Lucrezi.ro?
În fiecare zi oferte noi de muncă Puteți alege dintr-o gamă largă de locuri de muncă: Scopul nostru este de a oferi o gamă cât mai largă de opțiuni Lasă să-ți fie trimise noile oferte prin e-mail Fii primul care răspunde la noile oferte de muncă Toate ofertele de muncă într-un singur loc (de la angajatori, agenții și alte portaluri) Toate serviciile pentru persoanele aflate în căutarea unui loc de muncă sunt gratuite Vă vom ajuta să găsiți un nou loc de muncă