Senior Compiler Engineer - DL

About the role

NVIDIA is seeking a Deep Learning Compiler Engineer to join its DLC team. The DLC powers NVIDIA's inference engine, used in data centers, personal devices, automotive, and robotics.

Responsibilities

• Analyzing deep learning networks and developing compiler optimization algorithms.
• Collaborating with deep learning software framework teams and hardware architecture teams to accelerate deep learning software.
• Defining public APIs, performance optimizations, and analysis.
• Crafting and implementing compiler infrastructure techniques for neural networks.
• Performing general software engineering work.

Requirements

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

Qualifications

• Proficient in GPU and NPU architecture.
• Experience with systems with constrained resources, such as embedded platforms, small memory size, and cross compilation.
• Experience with MLIR, XLA, TVM, LLVM, deep learning models and algorithms, and deep learning frameworks like PyTorch.
• Experience mapping AI models to execute on custom NPU solutions.
• A track record of success in mentoring junior engineers and interns.

Skills

• Experience with GPU and NPU architecture.
• Experience with systems with constrained resources.
• Experience with MLIR, XLA, TVM, LLVM.
• Experience with deep learning models and algorithms.
• Experience with deep learning frameworks like PyTorch.
• Experience mapping AI models to execute on custom NPU solutions.
• Mentoring junior engineers and interns.

Benefits

With highly competitive salaries and a comprehensive benefits package, NVIDIA offers an inclusive work environment and is committed to fostering diversity in its workforce. The base salary range is $152,000 - $241,500 USD. Additionally, employees are eligible for equity and benefits.

Pay

Base salary range: $152,000 - $241,500 USD.

Schedule

Not specified.

Benefits

Comprehensive benefits package, including equity.