Senior Deep Learning Compiler Engineer
NVIDIA · Santa Clara, CA · 1 wk ago
EngineeringFull-time
About the role
NVIDIA is seeking a Deep Learning Compiler Engineer to join its DLC team. The DLC supports the inference engine across various platforms including data centers, personal devices, automotive, and robotics.
Responsibilities
- Analyze deep learning networks and develop compiler optimization algorithms.
- Collaborate with deep learning software framework and hardware architecture teams to accelerate deep learning software.
- Define public APIs, performance optimizations, and analysis for neural networks.
- Craft and implement compiler infrastructure techniques for neural networks.
- Conduct 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 required for working in a dynamic product-oriented team.
Qualifications
- Proficient in CPU and/or GPU architecture.
- CUDA or OpenCL programming experience.
- 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.
- GPU kernel generation with high performance and fast build time.
- Experience mentoring junior engineers and interns is a plus.
Benefits
Competitive salaries, comprehensive benefits package, and opportunities for equity. Applications accepted until May 4, 2026. NVIDIA is an equal opportunity employer committed to diversity and inclusion.