Staff Software Engineer, LLM Inference Platform at Zenlayer, Inc
United States
$300k - $600k
<p><span style="font-weight: bold"><u>About Zenlayer</u></span></p> <p>Zenlayer is the first massively distributed edge cloud provider, operating 290+ Points of Presence across 70+ countries. With accolades including PTC “Best Networking Innovation”, “Outstanding Cloud, Data Center, or Interconnection Company”, and Inc. 5000, we enable businesses to deploy and run applications at the edge, thus reducing network latency to the end user. Zenlayer focuses on improving connections worldwide, all while nurturing a unique company culture that values service, growth, and trust. As a “Zenplayer”, you’ll be joining a community of success-driven team players that share the dream of powering a better-connected world.</p> <p><br></p> <p><u>About the Role: </u></p> <p><span><span>As the <span style="font-weight: bold">technical lead for inference performance</span>, you will own end-to-end responsibility for the platform's Token Performance. For models ranging from hundreds of billions to trillions of parameters (including </span><span>MoE</span><span>), you will continuously reduce </span></span><span style="font-weight: bold"><span><span>cost per million tokens</span></span></span><span><span> and increase </span></span><span style="font-weight: bold"><span><span>per-GPU Goodput</span></span></span><span><span> while meeting TTFT/</span><span>TPOT SLOs, and</span><span> prove the gain from every optimization with reproducible data.</span></span><span> </span></p> <p><br><br></p> <p><span style="font-weight: bold"><span><span>What You Will Work On</span></span></span><span> </span></p> <p><span style="font-weight: bold"><span><span>1. End-to-End Performance Attribution and Benchmarking</span></span></span><span> </span></p> <ul> <li><span><span>Establish a unified Token Performance metrics framework (TTFT, TPOT, Goodput, GPU </span><span>utilization</span><span>, per-token cost), and build load-testing benchmarks that reflect real production traffic (high concurrency, long context, multi-turn Agent workloads)</span></span><span> </span></li> </ul> <ul> <li><span><span>Perform bottleneck attribution for models such as DeepSeek, Kimi, GLM, Qwen, and </span><span>MiniMax</span><span> across compute, memory bandwidth, communication, scheduling, and queuing, and let the data </span><span>determine</span><span> which layer is the dominant constraint</span></span><span> </span></li> </ul> <p><span style="font-weight: bold"><span><span>2. Kernel-Level Inference Engine Optimization</span></span></span><span> </span></p> <ul> <li><span><span>Dive deep into the source code of </span><span>vLLM</span><span> / </span><span>SGLang</span><span> / </span><span>TensorRT</span><span>-LLM, customizing and contributing upstream on critical paths such as the scheduler, Continuous Batching, Chunked Prefill, CUDA Graph, and KV block management</span></span><span> </span></li> </ul> <ul> <li><span><span>Clearly define the responsibility boundary between the inference engine and orchestration layers such as Dynamo / </span><span>llm</span><span>-d, and reduce tail latency through request routing and load balancing</span></span><span> </span></li> </ul> <p><span style="font-weight: bold"><span><span>3. PD Disaggregation and KV Cache Transfer</span></span></span><span> </span></p> <ul> <li><span><span>Lead the core design of Prefill-Decode disaggregation: the KV transfer path (NIXL/RDMA/UCX), transfer-compute overlap, and phase-aware autoscaling</span></span><span> </span></li> </ul> <ul> <li><span><span>Define the failure domains and recovery semantics of cross-instance KV transfer, and answer the key question: under which workloads </span><span>does</span><span> PD disaggregation deliver gains, and under which does it make things slower</span></span><span> </span></li> </ul> <p><span style="font-weight: bold"><span><span>4. </span><span>MoE</span><span> Parallelism and Wide-EP</span></span></span><span> </span></p> <ul> <li><span><span>Design hybrid TP/PP/DP/EP parallelism strategies and quantify the trade-offs between EP and TP</span></span><span> </span></li> </ul> <ul> <li><span><span>Tackle expert load imbalance and All-to-All communication overhead, and drive Wide-EP into production</span></span><span> </span></li> </ul> <p><span style="font-weight: bold"><span><span>5. Kernels and Quantization</span></span></span><span> </span></p> <ul> <li><span><span>Write and tune critical CUDA (and </span><span>ROCm</span><span>/HIP) operators: memory access optimization, kernel fusion, and Attention / </span><span>MoE</span><span> / communication-related kernels</span></span><span> </span></li> </ul> <ul> <li><span><span>Build a closed-loop "quantization – accuracy – throughput" evaluation: the real-world gains and accuracy regressions of FP8/FP4/INT4/AWQ/GPTQ across different engines and hardware</span></span><span> </span><br><br></li> </ul> <p><span style="font-weight: bold"><span><span>Requirements</span></span></span><span> </span></p> <ul> <li><span><span>First-principles understanding of Transformer inference: the compute vs. memory-bound nature of Prefill and Decode, the KV Cache memory model, the differences among MHA/GQA/MLA, and how </span><span>MoE</span><span> routing constrains parallelism</span></span><span> </span></li> </ul> <ul> <li><span><span>Source-level, in-depth experience with at least one mainstream inference engine, with real cases of </span><span>identifying</span><span> and fixing bottlenecks in scheduling, kernel </span><span>selection</span><span>, or KV management</span></span><span> </span></li> </ul> <ul> <li><span><span>CUDA kernel tuning skills (Nsight profiling, memory optimization, fusion, CUDA Graph) with hands-on practical experience</span></span><span> </span></li> </ul> <ul> <li><span><span>Bachelor’s degree in </span></span><span><span>Computer</span></span><span><span> Scienc</span></span><span><span>e, El</span></span><span><span>ectrical/Computer Engineering, or related field (HPC systems, ML Systems).</span></span><span> </span></li> </ul> <ul> <li><span><span>Minimum 6 years in systems/Performance Engineering, with at least 2-3 years directly on LLM </span></span><span><span>or GPU performance work</span></span><span> </span></li> </ul> <p><br></p> <p><span><span style="font-weight: bold"><span>Preferred, but not </span></span></span><span><span style="font-weight: bold"><span>required</span></span></span><span><span style="font-weight: bold"><span>: </span></span></span><span> </span></p> <ul> <li><span><span>Experience with </span></span><span><span>Merged PRs in </span></span><span><span>vLLM</span></span><span> </span></li> </ul> <ul> <li><span><span>Experience </span></span><span><span>operating</span></span><span><span> at cluster sale (</span></span><span><span>hundre</span></span><span><span>d</span></span><span><span>s</span></span><span><span> to thousands of GPUs)</span></span><span> </span></li> </ul>
Apply Now