Senior AI Compiler Engineer | ML-Driven GPU Optimizations + Equity

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

Benefits offered by this job

Equity

Generous benefits package

Job summary

2100 NVIDIA USA is seeking an AI Research Engineer / Applied Scientist to develop innovative technologies in machine learning compilers and AI systems. The ideal candidate will have strong expertise in AI/ML, particularly in compiler optimization and low-level programming, along with 5+ years of relevant experience.

This position offers a competitive salary range of 152,000 USD to 241,500 USD, equity, and benefits. The role is pivotal for advancing NVIDIA's commitment to technological innovation.

Qualifications

• 5+ years of experience building AI/ML systems.
• Strong software engineering skills in Python and at least one systems language.
• Hands-on experience training/fine-tuning large models.

Responsibilities

• Design and implement AI-based technology for GPU programming.
• Build training pipelines for supervised fine-tuning and reinforcement learning.
• Collaborate with compiler engineers to integrate learned policies.

Skills

Python

C++

AI/ML systems

Reinforcement learning

Prompt engineering

Education

M.S./PhD in Computer Engineering or Computer Science

Tools

CUDA

NVIDIA NeMo framework

Job description

2100 NVIDIA USA is seeking an AI Research Engineer / Applied Scientist to develop innovative technologies in machine learning compilers and AI systems. The ideal candidate will have strong expertise in AI/ML, particularly in compiler optimization and low-level programming, along with 5+ years of relevant experience.

This position offers a competitive salary range of 152,000 USD to 241,500 USD, equity, and benefits. The role is pivotal for advancing NVIDIA's commitment to technological innovation.