N

NVIDIA

Designs the chips and software that power AI

Inference Performance Engineer - Agent Driven Inference Optimization

Applied Machine LearningPosted on Nov 10
Full-timeRemote

About the Job

Requirements

  • A BS, MS, or PhD in Computer Science, Computer Engineering, Electrical Engineering, Applied Math, or a related field, or equivalent experience.
  • At least 3 years of relevant engineering experience.
  • Extensive knowledge of AI model execution efficiency and optimization, including continuous batching, throughput-latency tradeoffs, KV cache and memory limitations, parallel processing techniques, MoE serving, quantization, and serving SLAs.
  • Hands-on experience benchmarking and profiling GPU workloads with tools such as Nsight Systems, Nsight Compute, CUPTI, or PyTorch profiler, and interpreting kernel-level performance data.
  • Strong Python engineering skills and ability to navigate and modify large C++/CUDA serving codebases.
  • Rigorous experimental methodology using controlled single-variable comparisons, reproducible benchmarks, and evidence-backed optimization decisions.
  • Ability to explain performance tradeoffs clearly in writing and verbally to people and in documentation for autonomous systems.

Responsibilities

  • Distill performance instincts into reusable skills, workflows, and evidence-backed methodologies that AI agents can complete autonomously.
  • Review agent-generated experiments, validate findings, and curate best-known configurations.
  • Improve AI inference workload performance to increase throughput per GPU and user interactivity by exploring configuration options, parallelism techniques, batching, KV cache handling, quantization, and speculative decoding settings.
  • Measure and optimize aggregated and disaggregated serving architectures across TensorRT-LLM, SGLang, vLLM, and Dynamo on NVIDIA GPU platforms.
  • Profile workloads using Nsight Systems, kernel traces, and internal analysis tools; use roofline and speed-of-light analysis to identify performance headroom and drive fixes from hypothesis through measured results.
  • Contribute serving framework patches, optimized kernels, and deployment recipes upstream while maintaining model correctness.
  • Collaborate with TensorRT-LLM, SGLang, vLLM, kernel, benchmarking, and GPU architecture teams to turn profiling insights into delivered performance improvements.

Desired Qualifications

  • Direct contributions to TensorRT-LLM, vLLM, SGLang, FlashInfer, Dynamo, or comparable inference frameworks.
  • Experience with disaggregated serving, wide expert-parallel MoE inference, KV cache transfer, or NCCL/NIXL/NVSHMEM communication at multi-node scale.
  • CUDA kernel authorship or optimization experience on Hopper or Blackwell architectures, focusing on Tensor Cores, TMA, and warp specialization.
  • Proven results on public inference benchmarks such as MLPerf Inference or SemiAnalysis InferenceX.
  • Experience building or operating agentic AI workflows to automate engineering tasks.

About the Company

NVIDIA

Designs the chips and software that power AI

NVIDIA designs the chips and software that power artificial intelligence and sells complete data center systems to cloud providers, AI labs, and other businesses, along with GeForce graphics cards for gamers. Its GPUs, chips that perform thousands of calculations at once, became the main hardware for AI after researchers used them to train the AlexNet image-recognition model in 2012. NVIDIA connects many GPUs with its own processors and networking gear into servers and rack-sized computers, and developers program them through its CUDA software. These systems bring in most of its sales, and it also makes chips for cars and robots. Jensen Huang co-founded NVIDIA in 1993, and it is based in Santa Clara, California.

Company size
10,001+
Company stage
IPO
Headquarters
Santa Clara, CA
Founded
1993

Recently Posted Jobs

Explore more opportunities at NVIDIA

See more jobs