HyperGef: A Framework Enabling Efficient Fusion for Hypergraph Neural Network on GPUs. - Zhongming Yu; Guohao Dai; Shang Yang; Genghan Zhang; Hengrui Zhang; Feiwen Zhu; June Yang; Jishen Zhao; Yu Wang 002 (MLSys, 2023)

Canvas: End-to-End Kernel Architecture Search in Neural Networks. - Chenggang Zhao; Genghan Zhang; Mingyu Gao (CoRR, 2023)

Sgap: towards efficient sparse tensor algebra compilation for GPU. - Genghan Zhang; Yuetong Zhao; Yanting Tao; Zhongming Yu; Guohao Dai; Sitao Huang; Yuan Wen; Pavlos Petoumenos; Yu Wang 002 (CCF Trans. High Perform. Comput., 2023)

FEASTA: A Flexible and Efficient Accelerator for Sparse Tensor Algebra in Machine Learning. - Kai Zhong; Zhenhua Zhu; Guohao Dai; Hongyi Wang; Xinhao Yang; Haoyu Zhang; Jin Si; Qiuli Mao; Shulin Zeng; Ke Hong; Genghan Zhang; Huazhong Yang; et al. (ASPLOS (3), 2024)

GeoT: Tensor Centric Library for Graph Neural Network via Efficient Segment Reduction on GPU. - Zhongming Yu; Genghan Zhang; Hanxian Huang; Xin Chen; Jishen Zhao (CoRR, 2024)

CATS: Contextually-Aware Thresholding for Sparsity in Large Language Models. - Je-Yong Lee; Donghyun Lee; Genghan Zhang; Mo Tiwari; Azalia Mirhoseini (CoRR, 2024)

MoA: Mixture of Sparse Attention for Automatic Large Language Model Compression. - Tianyu Fu 004; Haofeng Huang; Xuefei Ning; Genghan Zhang; Boju Chen; Tianqi Wu; Hongyi Wang; Zixiao Huang; Shiyao Li; Shengen Yan; Guohao Dai; Huazhong Yang; et al. (CoRR, 2024)

Learning to (Learn at Test Time): RNNs with Expressive Hidden States. - Yu Sun 020; Xinhao Li; Karan Dalal; Jiarui Xu; Arjun Vikram; Genghan Zhang; Yann Dubois; Xinlei Chen; Xiaolong Wang 004; Sanmi Koyejo; Tatsunori Hashimoto; Carlos Guestrin (CoRR, 2024)

Compilation of Modular and General Sparse Workspaces. - Genghan Zhang; Olivia Hsu; Fredrik Kjolstad (Proc. ACM Program. Lang., 2024)

Mixture-of-Mamba: Enhancing Multi-Modal State-Space Models with Modality-Aware Sparsity. - Weixin Liang; Junhong Shen; Genghan Zhang; Ning Dong; Luke Zettlemoyer; Lili Yu (CoRR, 2025)

Adaptive Self-improvement LLM Agentic System for ML Library Development. - Genghan Zhang; Weixin Liang; Olivia Hsu; Kunle Olukotun (CoRR, 2025)