Senior AI Inference Compiler Engineer

NVIDIA Corporation; United States; Remote; USD 152,000 - 242,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA Corporation in the United States is seeking an AI & Deep Learning Compiler Engineer to join the DLC team. You will analyze networks, develop compiler optimizations, and collaborate with DL framework and GPU architecture teams to accelerate AI workloads on future NVIDIA GPUs.

You will define APIs, optimize performance, and contribute to both Ahead-of-Time and Just-in-Time compilation strategies, with equity and benefits on offer.

Qualifications

• Bachelor’s, Master’s or Ph.D. in Computer Science, Computer Engineering, related field or equivalent experience.
• 3+ years of relevant work or research experience in performance analysis and compiler optimizations.
• Experience with compiler technologies (e.g., MLIR, LLVM, XLA, Triton, etc.).
• Excellent C/C++ and Python programming and software design skills, including debugging, performance analysis, and test design.
• Ability to work independently, define project goals and scope, and lead your own development efforts.
• Strong interpersonal skills are required along with the ability to work in a dynamic product-oriented team.

Responsibilities

• Analyzing deep learning networks and developing compiler optimization algorithms.
• Collaborating with members of the deep learning software framework teams and the GPU architecture teams to accelerate the next generation of deep learning software.
• Scope of these efforts includes defining public APIs, performance optimizations and analysis, crafting and implementing compiler techniques for AI workloads and future NVIDIA GPUs.

Skills

C/C++ programming

Python programming

Performance analysis

Debugging

Software design

Education

Bachelor’s/Master’s/PhD in CS/CE or related field

Tools

MLIR

LLVM

XLA

Triton

Job description

NVIDIA Corporation in the United States is seeking an AI & Deep Learning Compiler Engineer to join the DLC team. You will analyze networks, develop compiler optimizations, and collaborate with DL framework and GPU architecture teams to accelerate AI workloads on future NVIDIA GPUs.

You will define APIs, optimize performance, and contribute to both Ahead-of-Time and Just-in-Time compilation strategies, with equity and benefits on offer.