← Search

Youhe Jiang

10 accepted papers

2026

Cascadia: An Efficient Cascade Serving System for Large Language Models

ICLR 2026poster

Recent advances in large language models (LLMs) have intensified the need to deliver both rapid responses and high-quality outputs. More powerful models yield better results but incur higher inference latency, whereas smaller models are faster yet less capable. Recent work proposes balancing this la…

Cited by 6SourceScholar
2026

Efficient Multi-round LLM Inference over Disaggregated Serving

ICML 2026poster

With the rapid evolution of Large Language Models (LLMs), multi-round workflows, such as autonomous agents and iterative retrieval, have become increasingly prevalent. However, this raises hurdles for serving LLMs under prefill-decode (PD) disaggregation, a widely adopted paradigm that separates the…

Cited by 0SourceScholar
2026

FSA: An Alternative Efficient Implementation of Native Sparse Attention Kernel

ICLR 2026poster

Recent advance in sparse attention mechanisms has demonstrated strong potential for reducing the computational cost of long-context training and inference in large language models (LLMs). Native Sparse Attention (NSA), one state-of-the-art approach, introduces natively trainable, hardware-aligned sp…

Cited by 0SourcecodeScholar
2026

HexGen-3: A Fully Disaggregated LLM Serving Framework with Fine-Grained Heterogeneous Resource Autoscaling

ICML 2026poster

The operational cost of serving large language models remains prohibitively high, largely due to extreme workload heterogeneity in production traffic. We observe that combining disaggregated inference with resource autoscaling enables fine-grained resource adjustment, allowing inference phases and o…

Cited by 0SourceScholar
2026

OServe: Accelerating LLM Serving via Spatial-Temporal Workload Orchestration

ICML 2026poster

Serving Large Language Models (LLMs) can benefit immensely from parallelizing both the model and input requests across multiple devices, but incoming workloads exhibit substantial spatial and temporal heterogeneity. Spatially, workloads comprise heterogeneous requests with varying compute and memory…

Cited by 0SourceScholar
2025

Demystifying Cost-Efficiency in LLM Serving over Heterogeneous GPUs

ICML 2025poster

Recent advancements in Large Language Models (LLMs) have led to increasingly diverse requests, accompanied with varying resource (compute and memory) demands to serve them. However, this in turn degrades the cost-efficiency of LLM serving as common practices primarily rely on homogeneous GPU resourc…

Cited by 4SourcePDFScholar
2025

Efficient Pre-Training of LLMs via Topology-Aware Communication Alignment on More Than 9600 GPUs

NeurIPS 2025poster

The scaling law for large language models (LLMs) depicts that the path towards machine intelligence necessitates training at large scale. Thus, companies continuously build large-scale GPU clusters, and launch training jobs that span over thousands of computing nodes. However, LLM pre-training prese…

Cited by 0SourceScholar
2025

HexGen-2: Disaggregated Generative Inference of LLMs in Heterogeneous Environment

ICLR 2025poster

Disaggregating the prefill and decoding phases represents an effective new paradigm for generative inference of large language models (LLM). This approach offers some significant system advantages, such as eliminating prefill-decoding interference and optimizing resource allocation. However, it is s…

Cited by 2SourcePDFScholar
2024

HexGen: Generative Inference of Large Language Model over Heterogeneous Environment

ICML 2024poster

Serving generative inference of the large language model is a crucial component of contemporary AI applications. In this paper, our focus lies in deploying such services in a heterogeneous and cross-datacenter setting to mitigate the substantial inference costs typically associated with a single cen…

2023

OSDP: Optimal Sharded Data Parallel for Distributed Deep Learning

IJCAI 2023poster

Large-scale deep learning models contribute to significant performance improvements on varieties of downstream tasks. Current data and model parallelism approaches utilize model replication and partition techniques to support the distributed training of ultra-large models. However, directly deployin…