Summary
NVIDIA is a technology company advancing accelerated computing and artificial intelligence through its GPU and software platforms. The company is seeking an AI Compiler Engineer to develop machine learning compiler and systems technologies, optimize AI workloads, build agent-driven development solutions, and collaborate with software and hardware engineering teams.
Responsibilities
- Innovate and develop new machine learning compiler and systems technologies
- Design, implement, and optimize compilers for high impact AI workloads
- Build agent driven solutions to automate the overall development flow
- Co-design agent system solutions with current and future AI compiler technologies
- Collaborate closely with other engineering teams at NVIDIA to build high impact solutions for machine learning acceleration
Skills
- * Bachelor's degree in Computer Science, Electrical Engineering, or related field (or equivalent experience)
- MS or PhD are preferred
- * Experience in machine learning systems development – including ML compilers, LLM inference kernels, kernel generations
- * Strong python, C/C++, GPU programming skills
- * Expertise in machine learning compilers (e.g. Apache TVM, MLIR)
- * Background in developing AI agent driven solutions and automation
- * Expertise in domain specific compiler and library solutions for LLM inference and training (e.g. FlashInfer, Flash Attention)
- * Strong experience machine learning systems research and productization
- * Open source project ownership or contributions
Qualifications
Must Haves
- * Bachelor's degree in Computer Science, Electrical Engineering, or related field (or equivalent experience)
Nice to Haves
- MS or PhD are preferred
- * Experience in machine learning systems development – including ML compilers, LLM inference kernels, kernel generations
- * Strong python, C/C++, GPU programming skills
- * Expertise in machine learning compilers (e.g. Apache TVM, MLIR)
- * Background in developing AI agent driven solutions and automation
- * Expertise in domain specific compiler and library solutions for LLM inference and training (e.g. FlashInfer, Flash Attention)
- * Strong experience machine learning systems research and productization
- * Open source project ownership or contributions
Benefits