Yikai Zhu's Blog

仰望星空，脚踏实地。 Looking up at the stars, with feet on the ground.

Yikai Zhu

Yikai Zhu

机器学习系统 · GPU Kernel · 生成式 AI Machine Learning Systems · GPU Kernels · Generative AI

欢迎来到 Yikai Zhu 的博客。我主要写机器学习系统、GPU / Triton kernel、推理框架与生成式模型相关的技术笔记。 这里的文章大多提供中文和英文两个版本，点击右上角的按钮即可切换语言。

Welcome to Yikai Zhu's blog. I write technical notes on machine learning systems, GPU / Triton kernels, inference frameworks, and generative models. Most posts are available in both Chinese and English — use the toggle in the top-right corner to switch languages.

论文Publications

  1. SpecForge: A Flexible and Efficient Open-Source Training Framework for Speculative Decoding
    S. Li, C. Wang, Y. Zhu, Y. Wang, F. Yin, S. Shi, Y. Chen, X. Dong, Q. Chen, et al.
    ICML 2026
    Code 维护者Maintainer
  2. SimAI: Unifying Architecture Design and Performance Tuning for Large-Scale Large Language Model Training with Scalability and Precision
    X. Wang, Q. Li, Y. Xu, G. Lu, D. Li, L. Chen, H. Zhou, L. Zheng, S. Zhang, Y. Zhu, et al.
    USENIX NSDI 2025
  3. Evolution of Aegis: Fault Diagnosis for AI Model Training Service in Production
    J. Dong, K. Qian, P. Zhang, Z. Zheng, L. Chen, F. Feng, Y. Xu, Y. Zhu, G. Lu, et al.
    USENIX NSDI 2025
  4. The Measure of All Measures: Quantifying LLM Benchmark Quality
    J. Yao, P. Jin, K. Bao, Q. Yu, K. Bhardwaj, C. Su, J. Wang, Y. Zhu, S. Devare, et al.
    NeurIPS 2025 Workshop
  5. Boosting Large-Scale Parallel Training Efficiency with C4: A Communication-Driven Approach
    J. Dong, B. Luo, J. Zhang, P. Zhang, F. Feng, Y. Zhu, A. Liu, Z. Chen, Y. Shi, et al.
    IEEE HPCA 2025
  6. Learning to Answer Ambiguous Questions with Knowledge Graph
    Y. Zhu, J. Shen, M. Zhang
    arXiv:1912.11668
    arXiv
  7. NoiGAN: Noise Aware Knowledge Graph Embedding with GAN
    K. Cheng, Y. Zhu, M. Zhang, Y. Sun