NVIDIA is looking for a Senior Software Engineer to develop groundbreaking technologies in the inference systems software stack. You will build innovative AI systems software to accelerate AI inference and develop libraries, code generators, and GPU kernel technologies for NVIDIA's hardware architecture.

What You'll Do

Innovate and develop new AI systems technologies for efficient inference
Design, implement, and optimize kernels for high impact AI workloads
Design and implement extensible abstractions for LLM serving engines
Build efficient just-in-time domain specific compilers and runtimes
Collaborate closely with other engineers at NVIDIA across deep learning frameworks, libraries, kernels, and GPU arch teams
Contribute to open source communities like FlashInfer, vLLM, and SGLang

What We're Looking For

Masters degree in Computer Science, Electrical Engineering, or related field (or equivalent experience)
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) and ideally inference engines and runtimes such as vLLM, SGLang, and MLC
Strong Python and C/C++ programming skills
Strong experience in GPU kernel development and performance optimizations (especially using CUDA C/C++, cuTile, Triton, or similar)

Nice to Have

PhD are preferred
Background in domain specific compiler and library solutions for LLM inference and training (e.g. FlashInfer, Flash Attention)
Expertise in inference engines like vLLM and SGLang
Expertise in machine learning compilers (e.g. Apache TVM, MLIR)
Open source project ownership or contributions

Technical Stack

PyTorch, JAX, TensorFlow, ONNX, vLLM, SGLang, MLC, Python, C/C++, CUDA C/C++, cuTile, Triton, FlashInfer, Flash Attention, Apache TVM, MLIR

Team & Environment

Collaborate closely with other engineers at NVIDIA across deep learning frameworks, libraries, kernels, and GPU arch teams.

Benefits & Compensation

Compensation: $184,000 USD - $287,500 USD + equity: Eligible
Equity
Benefits

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.