Responsibilities
- Innovate and develop new machine learning compiler and systems technologies
- Design, implement, and optimize compilers for high impact AI workloads
- Building strong kernel and domain specific language solutions for state of art kernels in LLM inference workloads
- Developing AI-driven solutions to automate the overall development flow.
- Co-design learning system solutions with current and future ML compiler and algorithm 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)
- 4+ years (academic/ industry) experience in machine learning systems development – including ML compilers, LLM inference kernels, kernel generations.
- Strong experience in developing or using deep learning frameworks (e.g. PyTorch, JAX etc)
- Strong python and C/C++ programming skills
Nice to Have
- Expertise in AI frameworks such as PyTorch, TensorFlow, and ONNX
- Expertise in machine learning compilers (e.g. Apache TVM, MLIR)
- Expertise in domain specific compiler and library solutions for LLM inference and training (e.g. FlashInfer, Flash Attention)
- Strong experience in GPU performance optimizations as well as experience machine learning systems research and productization
- Open source project ownership or contributions
Benefits
- equity
- benefits
Additional Information
- Applications for this job will be accepted at least until March 2, 2026.
- NVIDIA uses AI tools in its recruiting processes.
- NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.
