Senior Staff Inference Performance Engineer
d-Matrix
- United States
- Lead/Staff
- 8+ years
- Posted
Skills
Required
- LLM
- AI
- LLM serving
- Python
- C/C++
- Linux internals
- NUMA
- CPU pinning
- cpusets
- PCIe
- memory and cache behavior
- network paths
- Kubernetes
Preferred
- KV-cache
- speculative decoding
- profiling
- hardware
- LLM inference serving
- batching
- prefill / decode
- quantization
- GPU programming
- CUDA
Join d-Matrix, a leading AI company, as a Senior Staff Inference Performance Engineer. In this role, you will be responsible for the end-to-end performance of our inference workloads, ensuring they perform at least as well as they do on bare metal.
Responsibilities
- You will diagnose performance problems across the full serving path, isolate each problem to a specific layer and owner, and protect throughput, latency, and stability targets for orchestrated workloads.
- You will also assess what it takes to scale from single deployments to fleet operation and build lightweight telemetry and benchmarks.
- This position offers equity, health care, flexible time-off, paid paternity leave, a 401k retirement plan, and a remote & hybrid working model.
- Posséder la performance de bout en bout des charges de travail déployées sur la plateforme orchestrée, en veillant à ce qu'elles fonctionnent au moins aussi bien que sur du matériel nu.
- Diagnostiquer les problèmes de performance sur l'ensemble du chemin de service, de l'orchestration aux internes d'inférence, en isolant chaque problème à une couche spécifique.
- Écrire des déclarations de problème minimales et reproductibles pour les constatations inter-couches afin que les équipes responsables puissent agir sans répéter l'enquête.
- Sound judgment about ownership, with examples of correctly handing off problems that were not yours to fix
Requirements
- Clear technical writing that hardware, firmware, or platform teams can act on without context from your part of the stack
- 8+ years of hands-on performance or systems engineering, including resolving hard regressions that crossed multiple layers of a stack
- Proficiency in Python and C/C++, sufficient to instrument, debug, and patch code across the stack
- Deep systems fundamentals, including Linux internals, NUMA, CPU pinning and cpusets, PCIe, memory and cache behavior, and network paths
- Experience running performance-sensitive workloads on Kubernetes, including identifying when scheduler defaults, network overlays, or orchestration abstractions cost latency or throughput, and closing the gap
- Track record of finding performance overhead through profiling in unfamiliar, sparsely documented systems, valued more than experience with any specific vendor stack or accelerator
- Fluency with AI and agentic tools for investigation, code search, log triage, and drafting instrumentation, with the discipline to verify output against real data
- Experience with LLM serving frameworks such as vLLM, SGLang, or TensorRT-LLM
- Hands-on experience building or tuning speculative decoding, disaggregated serving, or heterogeneous serving architectures
- Experience tuning KV-cache management, request and batch scheduling, or weight prefetch and residency in an LLM serving system
Preferred
- Working knowledge of LLM inference serving, including batching, KV-cache, prefill/decode, quantization, and speculative decoding
- GPU programming and profiling experience (such as CUDA, Nsight, NCCL), valued as transferable background
- Experience with pre-production or bring-up hardware, or in organizations where root causes routinely cross team boundaries
- Rust experience
- Startup experience or comfort in environments where priorities shift quickly
Pay
USD 175,000 - 265,000
jobapplier