AI Performance Engineer — Deep Learning & LLM Acceleration

NVIDIA Gruppe; Santa Clara (CA); On-site; USD 152,000 - 241,500; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA Gruppe is looking for software engineers at all experience levels to enhance Deep Learning models and libraries. As part of a diverse team, you will support the development of cutting-edge AI tools and architectures, working closely with the open-source community.

The ideal candidate possesses a strong background in C++ and Python, along with experience in parallel systems programming. A competitive salary range of 152,000USD – 241,500USD exists for Level3 candidates.

Qualifications

• 3+ years of experience in C++ and Python programming.
• Strong background in parallel systems programming, preferably on GPUs.
• Knowledge of Computer Architecture, Code Optimization, and Operating Systems.

Responsibilities

• Build and support Transformer Engine, the open-source library for accelerating training of Large Language Models.
• Collaborate on systems research to improve Deep Learning model performance.
• Implement, benchmark, and optimize new Deep Learning models to scale efficiently on NVIDIA GPUs.

Skills

C++ Programming

Python Programming

Parallel Systems Programming

Communication Skills

Education

Bachelor's degree or equivalent in Computer Science, Electrical Engineering, or a related field

Tools

PyTorch

JAX

CUDA

Job description

NVIDIA Gruppe is looking for software engineers at all experience levels to enhance Deep Learning models and libraries. As part of a diverse team, you will support the development of cutting-edge AI tools and architectures, working closely with the open-source community.

The ideal candidate possesses a strong background in C++ and Python, along with experience in parallel systems programming. A competitive salary range of 152,000USD – 241,500USD exists for Level3 candidates.