Senior AI Compiler Engineer, Algorithms and Code-Generation
$152k - $242k • US, CA, Santa Clara
Posted 1d ago
About the job
NVIDIA is seeking a Senior AI & Deep Learning Compiler Engineer to join its Deep Learning & AI Compiler (DLC) team. This role involves analyzing deep learning networks and developing compiler optimization algorithms to power the next era of AI computing. The DLC is a critical component for NVIDIA's inference engine, enabling breakthroughs across various AI domains. The compiler must deliver leading inference performance, fast build times, reduced memory footprints, and ease of use for both Ahead-of-Time and Just-in-Time compilation. You will be part of a team building a compiler used by the global deep learning community, impacting data centers, personal devices, automotive, and robotics.
Responsibilities
- Analyze deep learning networks and develop compiler optimization algorithms.
- Define public APIs, performance optimizations, and analysis.
- Craft and implement compiler techniques for AI workloads and future NVIDIA GPUs.
- Analyze and debug performance bottlenecks on GPUs using CUDA.
Requirements
- Bachelor's, Master's, or Ph.D. in Computer Science, Computer Engineering, or 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
- Highly competitive salaries
- Comprehensive benefits package
- Equity
- Eligibility for equity and benefits