Note: The job is a remote job and is open to candidates in USA. NVIDIA is an AI computing company developing GPU technologies that power deep learning and modern computing. The company is seeking a Senior AI Compiler Engineer to build an MLIR-based AI compiler for NVIDIA’s inference engine, focusing on performance, build efficiency, memory use, and usability across data center and edge environments.
Responsibilities
- Develop MLIR-based graph representations and optimizations for future GPU architectures
- Partner with framework and hardware teams to enable new model patterns and upcoming GPU architectural features
- Define APIs and MLIR dialects, conduct performance optimizations and analysis, implement compiler optimizations and kernel generation for neural networks, and contribute to other general software engineering work
Skills
- Bachelor's, Master's, or Ph.D. 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
- Experience with compiler technologies such as MLIR, XLA, and LLVM
- Excellent C/C++ and Python programming and software design skills, including debugging, performance analysis, and testing
- Ability to work independently, define project goals and scope, and lead your own development efforts
- Strong interpersonal skills and the ability to thrive in a fast-moving, dynamic, product-oriented team
- Understanding of deep learning models, algorithms, and frameworks such as PyTorch and JAX
- Experience with GPU kernel generation targeting high performance and fast build times
- Proficiency in GPU architecture with CUDA or OpenCL programming experience
- A track record of mentoring early career engineers and interns is a bonus
Benefits
- You will also be eligible for equity
- Benefits
Company Overview
Company H1B Sponsorship