Hybrid JIT-CUDA Graph Optimization for Low-Latency Large Language Model Inference
Abstract
Large Language Models (LLMs) deliver strong performance but suffer from high inference latency and unstable kernel-launch overhead. This the- sis introduces a Hybrid JIT–CUDA Graph Runtime that combines static CUDA Graph replay with JIT-compiled dynamic kernels to achieve deter- ministic, low-latency autoregressive decoding. A rolling graph-generation mechanism enables variable sequence lengths while preserving static-shape constraints. Evaluated on an NVIDIA H100 GPU, the system reduces Time- to-First-Token (TTFT) by 45–80%, achieves the lowest and most stable P99 latency, and provides the fastest end-to-end 500-token generation compared with HuggingFace (PyTorch Eager) and TensorRT–LLM. The results demon- strate that integrating JIT flexibility with CUDA Graph determinism offers an effective approach for high-performance LLM inference.
Fields of Study & Subjects
Parallel Computing and Optimization Techniques
Natural Language Processing Techniques
Topic Modeling
Publication details
- Journal / source
- IEEE Annual International Computer Software and Applications Conference (COMPSAC)
- Pages
- 196–204
- Publication date
- 1 July 2026
- Language
- EN