Senior Dl Algorithms Engineer - Inference Performance

at Nvidia

USD 152,000-287,500 per year

SENIOR

✅ On-site

Tech Stack
Tag name is followed by "@" symbol and proficiency level value. About proficiency levels:

1-2 — basic awareness. Minimal hands-on experience, and a rudimentary understanding of the technology's purpose;

3-6 — daily use. Comfortable and regular usage, capable of handling common tasks and challenges related to the technology;

7-9 — you are an expert, you can teach others, you know all the pitfalls and tricks;

10 — exceptional knowledge, comprehensive understanding, and adeptness in all aspects of the technology, including advanced problem-solving. Think twice before claiming or demanding such level.

AI @ 6 Algorithms @ 7 CUDA @ 1 Deep Learning @ 7 GPU @ 4 HPC @ 6 LLM @ 7 OpenCL @ 1 Performance Analysis @ 4 Performance Optimization @ 4 Profiling @ 4 PyTorch @ 6 SGLang vLLM

Details

We are looking for a Senior DL Algorithms Engineer for LLM/Omni model optimizations. Seeking senior engineers who are mindful of performance analysis and optimization to help us squeeze every last clock cycle out of Deep Learning workloads. If you are unafraid to work across all layers of the hardware/software stack from GPU architecture to Deep Learning Framework to achieve peak performance, we want to hear from you. This role offers an opportunity to directly impact the hardware and software roadmap in a fast-growing technology company that leads the AI revolution.

Responsibilities

Enable and optimize state-of-the-art open models (like Nemotron and Cosmos) on NVIDIA’s accelerated inference SW stack.
Contribute new features, fix bugs and deliver production code to open-source frameworks like TRT-LLM, vLLM, SGLang, FlashInfer, etc.
Profile and analyze bottlenecks across the full inference stack to push the boundaries of inference performance.
Benchmark state-of-the-art offerings and perform competitive analysis for NVIDIA’s SW/HW stack.
Co-design with partner teams to develop the next generation of AI models and services.

Requirements

PhD in CS, EE or CSEE or equivalent experience.
3+ years of experience.
Strong background in deep learning and neural networks, in particular inference.
Experience with performance profiling, analysis and optimization, especially for GPU-based applications.
Proficient in PyTorch or equivalent frameworks for AI, or HPC-heavy application development.
Deep understanding of computer architecture, and familiarity with the fundamentals of GPU architecture.

Ways to stand out from the crowd

Proven experience with processor and system-level performance optimization.
Deep understanding of modern LLM/Diffusion architectures.
Strong fundamentals in algorithms.
GPU programming experience (CUDA or OpenCL) is a strong plus.

Compensation

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 152,000 USD - 241,500 USD for Level 3, and 184,000 USD - 287,500 USD for Level 4.

You will also be eligible for equity and benefits.