Senior Deep Learning Compiler Engineer

at Nvidia

USD 152,000-241,500 per year

SENIOR

✅ On-site

Tech Stack
Tag name is followed by "@" symbol and proficiency level value. About proficiency levels:

1-2 — basic awareness. Minimal hands-on experience, and a rudimentary understanding of the technology's purpose;

3-6 — daily use. Comfortable and regular usage, capable of handling common tasks and challenges related to the technology;

7-9 — you are an expert, you can teach others, you know all the pitfalls and tricks;

10 — exceptional knowledge, comprehensive understanding, and adeptness in all aspects of the technology, including advanced problem-solving. Think twice before claiming or demanding such level.

API Algorithms @ 4 CUDA @ 4 Debugging @ 6 Deep Learning @ 4 GPU @ 6 LLVM @ 4 Mentoring @ 4 OpenCL @ 4 Performance Analysis @ 6 PyTorch @ 4 Python @ 6 Robotics

Details

NVIDIA is hiring a Deep Learning Compiler Engineer for its Deep Learning Compiler team. The team develops compiler technology that powers NVIDIA's inference engine across data centers, personal devices, automotive, and robotics. The compiler is designed to deliver leading inference performance, fast build times, reduced memory footprints, and ease of use through both Ahead-of-Time and Just-in-Time compilation.

Responsibilities

Analyze deep learning networks and develop compiler optimization algorithms.
Collaborate with deep learning software framework teams and hardware architecture teams to accelerate the next generation of deep learning software.
Define public APIs, implement performance optimizations and analysis, and develop compiler infrastructure techniques for neural networks.
Perform general software engineering work related to deep learning compiler development.
Independently define project goals and scope and lead development efforts.

Requirements

Bachelor's, Master's, or Ph.D. degree in Computer Science, Computer Engineering, a related field, or equivalent experience.
3+ years of relevant work or research experience in performance analysis and compiler optimizations.
Excellent C/C++ and Python programming and software design skills, including debugging, performance analysis, and test design.
Ability to work independently and define project goals and scope.
Strong interpersonal skills and the ability to work in a dynamic, product-oriented team.

Preferred Qualifications

Proficiency in CPU and/or GPU architecture.
CUDA or OpenCL programming experience.
Experience with systems that have constrained resources, including embedded platforms, small memory sizes, and cross-compilation.
Experience with MLIR, XLA, TVM, LLVM, deep learning models and algorithms, and deep learning frameworks such as PyTorch.
Experience with GPU kernel generation focused on high performance and fast build times.
Experience mentoring junior engineers and interns.

Benefits

The base salary range is $152,000–$241,500 USD, determined by location, experience, and the pay of employees in similar positions. The role also includes eligibility for equity and benefits. NVIDIA is committed to fostering a diverse work environment and is an equal opportunity employer.