Senior Deep Learning Software Engineer, Inference

Requirements:

• 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.

What you'll be doing:

• 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.

Nice to haves:

• 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).

Perks and benefits:

• NVIDIA is at the forefront of breakthroughs in Artificial Intelligence, High-Performance Computing, and Visualization.
• Highly competitive salaries.
• Extensive benefits package.
• Work environment that promotes diversity, inclusion, and flexibility.
• Equal opportunity employer committed to fostering a supportive and empowering workplace for all.