Senior Deep Learning Software Engineer, Inference

Performance optimization, analysis, and tuning of DL models in various domains like LLM, Multimodal and Generative AI.

Scale performance of DL models across different architectures and types of NVIDIA accelerators.

Contribute features and code to NVIDIA’s inference libraries, vLLM and SGLang, FlashInfer and LLM software solutions.

Work with cross-collaborative teams across frameworks, NVIDIA libraries and inference optimization innovative solutions.

Masters or PhD or equivalent experience in relevant field (Computer Engineering, Computer Science, EECS, AI).

5+ years of relevant software development experience.

Excellent C/C++ programming and software design skills. SW Agile skills are helpful and Python experience is a plus.

Prior experience with training, deploying or optimizing the inference of DL models in production is a plus.

Prior background with performance modeling, profiling, debug, and code optimization or architectural knowledge of CPU and GPU is a plus.

Contribute to Deep Learning Software projects, such as PyTorch, vLLM, and SGLang to drive advancements in the field.

Experience with Multi-GPU Communications (NCCL, NVSHMEM)

Experience building and shipping products to enterprise customers.

GPU programming experience (CUDA, OAI TRITON or CUTLASS).

NVIDIA is at the forefront of breakthroughs in Artificial Intelligence, High-Performance Computing, and Visualization. Our teams are composed of driven, innovative professionals dedicated to pushing the boundaries of technology. We offer highly competitive salaries, an extensive benefits package, and a work environment that promotes diversity, inclusion, and flexibility. As an equal opportunity employer, we are committed to fostering a supportive and empowering workplace for all