Senior CUDA AI Frameworks Engineer

NVIDIA; Town of Texas (WI); On-site; USD 184,000 - 356,500; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is seeking a highly skilled individual for AI development focused on CUDA integration and performance optimization. Candidates should possess a BS, MS, or PhD in relevant fields, with 8+ years of experience in deep learning frameworks and CUDA.

This role involves working closely with other experts to innovatively improve AI workloads and systems. The base salary ranges from 184,000 USD to 356,500 USD, depending on level and experience. Inclusion and diversity are core to NVIDIA's values.

Qualifications

• 8+ years of relevant industry experience or equivalent academic experience after completed degree.
• Development experience with Deep Learning Frameworks such as PyTorch, JAX.
• Solid grasp of compiler technologies (e.g., torch.compile).

Responsibilities

• Integrate new CUDA features and Runtime abstractions in AI frameworks.
• Perform deep analysis of AI workloads to identify opportunities for innovation.
• Develop exploratory tools and runtime systems to profile and accelerate deep learning.

Skills

Deep Learning Frameworks

CUDA

Python

C++

AI models understanding

Education

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

Tools

Performance profiler tools

CUDA

Job description

NVIDIA is seeking a highly skilled individual for AI development focused on CUDA integration and performance optimization. Candidates should possess a BS, MS, or PhD in relevant fields, with 8+ years of experience in deep learning frameworks and CUDA.

This role involves working closely with other experts to innovatively improve AI workloads and systems. The base salary ranges from 184,000 USD to 356,500 USD, depending on level and experience. Inclusion and diversity are core to NVIDIA's values.