Senior AI Software Engineer, Kernel Libraries

We're looking for outstanding AI systems engineers to develop groundbreaking technologies in the inference systems software stack. As a member of the team, you'll develop libraries, code generators, and GPU kernel technologies for NVIDIA's hardware architecture.

Requirements

• Masters degree in Computer Science, Electrical Engineering, or related field (or equivalent experience); PhD are preferred
• 6+ years (academic/ industry) experience with ML/DL systems development preferable
• Strong experience in developing or using deep learning frameworks (e.g. PyTorch, JAX, TensorFlow, ONNX, etc) and ideally inference engines and runtimes such as vLLM, SGLang, and MLC.
• Strong Python and C/C++ programming skills

Benefits

• Eligible for equity
• Benefits offered