Senior DL Performance Engineer - LLM/Transformer Optimizations

NVIDIA AI; Santa Clara (CA); On-site; USD 152,000 - 288,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is seeking software engineers to join a Deep Learning models performance engineering team tasked with building and optimizing libraries and tools for efficient AI applications. The team accelerates DL frameworks like PyTorch and JAX, enabling researchers and engineers to design, develop, and deploy state-of-the-art AI.

We work across NVIDIA and the broader open-source community to deliver performance gains on the world’s leading AI platform.

Qualifications

• BS or equivalent experience in Computer Science, Electrical Engineering, or a related field.
• 3+ years of experience in C++ and Python programming.
• Strong background, experience, or coursework in parallel systems programming, preferably on GPUs.
• Knowledge of Computer Architecture, Code Optimization, and/or Operating Systems.
• Proven experience in developing large software projects.
• Excellent verbal and written communication skills.

Responsibilities

• Build and support Transformer Engine, the open-source library for accelerating the training of Large Language Models.
• Collaborate on systems research that improves Deep Learning model performance, such as training using extremely low precision, parallelism methods, etc.
• Implement, benchmark, and optimize new Deep Learning models such as LLMs straight out of groundbreaking research to scale efficiently on NVIDIA GPUs and systems.
• Build and contribute to NVIDIA submissions on community benchmarks such as MLPerf.
• Engage with the open-source community as well as support enterprise customers and partners by delivering the benefits of NVIDIA’s latest hardware and software innovations.
• Influence the design of new hardware generations and core platform software components for NVIDIA hardware and systems.

Skills

C++

Python

Parallel systems

GPU

Communication

Education

BS in Computer Science, Electrical Engineering, or related field

Tools

cuBLAS

cuDNN

cuSOLVER

CUDA

Triton

Job description

NVIDIA is seeking software engineers to join a Deep Learning models performance engineering team tasked with building and optimizing libraries and tools for efficient AI applications. The team accelerates DL frameworks like PyTorch and JAX, enabling researchers and engineers to design, develop, and deploy state-of-the-art AI.

We work across NVIDIA and the broader open-source community to deliver performance gains on the world’s leading AI platform.