Senior Compiler Engineer, AI Inference Platforms
$152k - $242k • Remote • US, CA, Santa Clara
Posted 1d ago
About the job
NVIDIA is seeking an AI & Deep Learning Compiler Engineer to join its Deep Learning & AI Compiler (DLC) team. This role involves developing and optimizing the compiler that powers NVIDIA's inference engine across various platforms, from data centers to personal devices. The compiler is crucial for delivering leading inference performance, fast build times, reduced memory footprints, and ease of use for the global deep learning community. You will contribute to the next era of computing by enabling breakthroughs in areas like large language models and generative AI.
Responsibilities
- Analyze deep learning networks and develop compiler optimization algorithms.
- Collaborate with deep learning software framework and GPU architecture teams to accelerate next-generation deep learning software.
- Define public APIs, conduct performance optimizations and analysis, and implement compiler techniques for AI workloads and future NVIDIA GPUs.
Requirements
- Bachelor’s, Master’s, or Ph.D. in Computer Science, Computer Engineering, or a related field, or equivalent experience.
- 3+ years of relevant work or research experience in performance analysis and compiler optimizations.
- Experience with compiler technologies such as MLIR, LLVM, XLA, or Triton.
- 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 development efforts.
- Strong interpersonal skills and ability to work in a dynamic, product-oriented team.
Benefits
- Competitive salaries
- Generous benefits package
- Equity