Senior CUDA DL Frameworks Engineer - Equity Options

Nvidia Corporation; Santa Clara (CA); On-site; USD 184,000 - 287,500; Full time

Benefits offered by this job

Eligible for equity

Comprehensive benefits package

Job summary

Nvidia Corporation is seeking an AI Compiler Engineer to innovate with CUDA features and deep learning frameworks. The role requires collaboration across teams to enhance performance in AI workloads and utilizes strong programming skills in Python and C++. Candidates should have extensive experience and a relevant degree.

The salary range is between $184,000 and $287,500 for Level 4, and up to $356,500 for Level 5, along with equity and benefits.

Qualifications

• 8+ years of relevant industry experience or equivalent academic experience.
• Experience with Deep Learning Frameworks such PyTorch, JAX.
• Solid understanding of HPC/AI communication concepts.

Responsibilities

• Integrate new CUDA features and AI frameworks.
• Perform deep analysis of AI workloads.
• Own and drive improvements in the AI Compiler-Runtime interface.

Skills

Development experience with Deep Learning Frameworks

Rapid prototyping with Python, C++, CUDA

Performance benchmarking on AI clusters

Solid grasp of AI models

Adaptability and passion for learning

Education

BS, MS, or PhD in Computer Science or related field

Tools

PyTorch

CUDA

NVIDIA Nsight Systems

Job description

Nvidia Corporation is seeking an AI Compiler Engineer to innovate with CUDA features and deep learning frameworks. The role requires collaboration across teams to enhance performance in AI workloads and utilizes strong programming skills in Python and C++. Candidates should have extensive experience and a relevant degree.

The salary range is between $184,000 and $287,500 for Level 4, and up to $356,500 for Level 5, along with equity and benefits.