Explanation Graph Generation via Generative Pre-training over Synthetic Graphs. - Han Cui; Shangzhan Li; Yu Zhang; Qi Shi (ACL (Findings), 2023)

TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operators. - Jianling Li; Shangzhan Li; Zhenye Gao; Qi Shi 002; Yuxuan Li; Zefan Wang; Jiacheng Huang; WangHaojie WangHaojie; Jianrong Wang; Xu Han 007; Zhiyuan Liu 001; Maosong Sun 001 (ACL (Findings), 2025)

TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operators. - Jianling Li; Shangzhan Li; Zhenye Gao; Qi Shi 002; Yuxuan Li; Zefan Wang; Jiacheng Huang; Haojie Wang; Jianrong Wang; Xu Han 007; Zhiyuan Liu 001; Maosong Sun 001 (CoRR, 2025)

AutoTriton: Automatic Triton Programming with Reinforcement Learning in LLMs. - Shangzhan Li; Zefan Wang; Ye He; Yuxuan Li; Qi Shi 002; Jianling Li; Yonggang Hu; Wanxiang Che; Xu Han 007; Zhiyuan Liu 001; Maosong Sun 001 (CoRR, 2025)