Description
NVIDIA is seeking outstanding AI compiler engineers to develop groundbreaking technologies in machine learning compilers and AI systems.
As an AI Compiler Engineer at NVIDIA, you will design, implement, and optimize compilers for high-impact AI workloads. You will work on innovative AI compiler solutions that integrate with NVIDIA's software stack to provide comprehensive acceleration for modern machine learning models.
Key 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
Requirements:
- 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
Preferred Qualifications:
- 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 in machine learning systems research and productization
- Open-source project ownership or contributions
Compensation:
Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is $108,000 - $178,250 for Level 1, and $124,000 - $195,500 for Level 2. You will also be eligible for equity and benefits.