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); MS or PhD are preferred
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

Benefits

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

machine learningcompiler developmentAI workloadsdeep learning frameworksPythonC++CLLM inferencekernel generationAI-driven solutions

Soft Skills

collaborationinnovationoptimizationdesignimplementation

Certifications

Bachelor's degreeMaster's degreePhD