FLuRKA: Fast fused Low-Rank & Kernel Attention. - Ahan Gupta; Yueming Yuan; Yanqi Zhou; Charith Mendis (CoRR, 2023)

SPLAT: A Framework for Optimised GPU Code-Generation for SParse reguLar ATtention. - Ahan Gupta; Yueming Yuan; Devansh Jain 001; Yuhao Ge; David Aponte; Yanqi Zhou; Charith Mendis (Proc. ACM Program. Lang., 2025)

MegaFold: System-Level Optimizations for Accelerating Protein Structure Prediction Models. - Hoa La; Ahan Gupta; Alex Morehead; Jianlin Cheng; Minjia Zhang (CoRR, 2025)

X-MoE: Enabling Scalable Training for Emerging Mixture-of-Experts Architectures on HPC Platforms. - Yueming Yuan; Ahan Gupta; Jianping Li; Sajal Dash; Feiyi Wang; Minjia Zhang (CoRR, 2025)