Senior Software Engineer, AI Inference Systems

NVIDIA; Lavamünd; Vor Ort; EUR 120.000 - 160.000; Vollzeit

Zusammenfassung

NVIDIA seeks highly skilled software engineers to build AI inference systems that scale models efficiently. You will architect and implement high-performance stacks, optimize GPU kernels and compilers, and drive benchmarks across multi-GPU, multi-node, and multi-cloud environments.

You will collaborate with inference, compiler, scheduling, and performance teams to push the frontier of accelerated AI computing while delivering practical, production-ready software across NVIDIA’s platforms.

Qualifikationen

• Bachelor’s degree in Computer Science, Computer Engineering or Software Engineering with 7+ years of experience; or Master’s 5+ years; or PhD with strong ML Systems publications.
• Strong programming skills in Python and C/C++; knowledge of Go or Rust is a plus.
• Deep fundamentals in algorithms, data structures, OS, architecture, parallel/distributed systems and DL theory.
• Familiarity with performance engineering in ML frameworks (e.g., PyTorch) and inference engines (e.g., vLLM).
• Experience with GPU programming, CUDA, memory hierarchies, and profiling tools (Nsight).

Aufgaben

• Contribute features to vLLM and optimize inference frameworks for new NVIDIA GPUs.
• Develop, optimize, and benchmark GPU kernels using fusion, autotuning, and memory/layout optimization.
• Define benchmarking methodologies and contribute to MLPerf Inference submissions.
• Architect scheduling and orchestration of containerized large-scale inference deployments on GPU clusters.
• Publish original research and translate ideas into NVIDIA software products.

Kenntnisse

Python

C/C++

Go or Rust

Algorithms & data structures

Parallel programming

Distributed systems

Performance engineering

GPU programming

CUDA

Nsight (Systems/Compute)

Containerization

Linux namespaces & cgroups

Communication skills

Ausbildung

Bachelor’s degree in CS/CE/SE

Master’s degree in CS/CE/SE

PhD in ML Systems / GPU architecture / HPC

Tools

TorchDynamo/Inductor

Triton

MLIR/LLVM

CUTLASS

XLA

ML tooling (Nsight)

Jobbeschreibung

We are seeking highly skilled and motivated software engineers to join us and build AI inference systems that serve large-scale models with extreme efficiency. You’ll architect and implement high-performance inference stacks, optimize GPU kernels and compilers, drive industry benchmarks, and scale workloads across multi-GPU, multi-node, and multi-cloud environments. You’ll collaborate across inference, compiler, scheduling, and performance teams to push the frontier of accelerated computing for AI.

What You’ll Be Doing

• Contribute features to vLLM that empower the newest models with the latest NVIDIA GPU hardware features; profile and optimize the inference framework (vLLM) with methods like speculative decoding, data/tensor/expert/pipeline-parallelism, prefill-decode disaggregation.
• Develop, optimize, and benchmark GPU kernels (hand-tuned and compiler-generated) using techniques such as fusion, autotuning, and memory/layout optimization; build and extend high-level DSLs and compiler infrastructure to boost kernel developer productivity while approaching peak hardware utilization.
• Define and build inference benchmarking methodologies and tools; contribute both new benchmark and NVIDIA’s submissions to the industry-leading MLPerf Inference benchmarking suite.
• Architect the scheduling and orchestration of containerized large-scale inference deployments on GPU clusters across clouds.
• Conduct and publish original research that pushes the pareto frontier for the field of ML Systems; survey recent publications and find a way to integrate research ideas and prototypes into NVIDIA’s software products.

What We Need To See

• Bachelor’s degree (or equivalent expeience) in Computer Science (CS), Computer Engineering (CE) or Software Engineering (SE) with 7+ years of experience; alternatively, Master’s degree in CS/CE/SE with 5+ years of experience; or PhD degree with the thesis and top-tier publications in ML Systems, GPU architecture, or high-performance computing.
• Strong programming skills in Python and C/C++; experience with Go or Rust is a plus; solid CS fundamentals: algorithms & data structures, operating systems, computer architecture, parallel programming, distributed systems, deep learning theories.
• Knowledgeable and passionate about performance engineering in ML frameworks (e.g., PyTorch) and inference engines (e.g., vLLM and SGLang).
• Familiarity with GPU programming and performance: CUDA, memory hierarchy, streams, NCCL; proficiency with profiling/debug tools (e.g., Nsight Systems/Compute).
• Experience with containers and orchestration (Docker, Kubernetes, Slurm); familiarity with Linux namespaces and cgroups.
• Excellent debugging, problem-solving, and communication skills; ability to excel in a fast-paced, multi-functional setting.

Ways to stand out from the crowd

• Experience building and optimizing LLM inference engines (e.g., vLLM, SGLang).
• Hands-on work with ML compilers and DSLs (e.g., Triton, TorchDynamo/Inductor, MLIR/LLVM, XLA), GPU libraries (e.g., CUTLASS) and features (e.g., CUDA Graph, Tensor Cores).
• Experience contributing to containerization/virtualization technologies such as containerd/CRI-O/CRIU.
• Experience with cloud platforms (AWS/GCP/Azure), infrastructure as code, CI/CD, and production observability.
• Contributions to open-source projects and/or publications; please include links to GitHub pull requests, published papers and artifacts.

At NVIDIA, we believe artificial intelligence (AI) will fundamentally transform how people live and work. Our mission is to advance AI research and development to create groundbreaking technologies that enable anyone to harness the power of AI and benefit from its potential. Our team consists of experts in AI, systems and performance optimization. Our leadership includes world-renowned experts in AI systems who have received multiple academic and industry research awards. If you’re excited to build systems, kernels, and tools that make large-scale AI faster, more efficient, and easier to deploy, we’d love to hear from you.

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. For Poland: The base salary range is 292,500 PLN - 507,000 PLN for Level 4, and 375,000 PLN - 650,000 PLN for Level 5. , , JR2017366