Author

Jia-Nian Zhu

h-index 2 View on Semantic Scholar

We have 2 of 5 papers

We haven’t gathered this author’s papers yet. Follow them and we’ll fetch their work.

Not the right person? Other researchers publish under this name.

#small language model Preprint Sep 2026

Epoch: Compiling Diffusion Blocks for Sparse MoE Serving

A serving system that treats the diffusion block as a compilation unit that improves end-to-end execution time by up to 2.7 times over the strongest surviving baseline under the same 8-GPU placement and remains feasible at the largest batch sizes where multiple baselines run out of memory, while preserving task quality...

Jia-Nian Zhu, Hang Wu, Ying-Hui Li et al. · 0 citations

Book Open access Jul 2026

Taming Dynamic Diffusion LLM Inference through Virtual Static Execution

Prism abstracts the highly dynamic diffusion workload into a predictable, static execution flow transparent to the compiler, and achieves this via three techniques: spatial regularization, temporal stabilization, and specialized kernels that selectively bypass padding data.

Jianian Zhu, Hang Wu, Yinghui Li et al. · International Conference on... · 0 citations