A Flexible Approach to Autotuning Multi-Pass Machine Learning Compilers. - Phitchaya Mangpo Phothilimthana; Amit Sabne; Nikhil Sarda; Karthik Srinivasa Murthy; Yanqi Zhou; Christof Angermueller; Mike Burrows; Sudip Roy 002; Ketan Mandke; Rezsa Farahani; Yu Emma Wang; Berkin Ilbeyi; et al. (PACT, 2021)

Overlap Communication with Dependent Computation via Decomposition in Large Deep Learning Models. - Shibo Wang; Jinliang Wei; Amit Sabne; Andy Davis; Berkin Ilbeyi; Blake Hechtman; Dehao Chen; Karthik Srinivasa Murthy; Marcello Maggioni; Qiao Zhang; Sameer Kumar; Tongfei Guo; et al. (ASPLOS (1), 2023)

TensorRight: Automated Verification of Tensor Graph Rewrites. - Jai Arora; Sirui Lu; Devansh Jain 001; Tianfan Xu; Farzin Houshmand; Phitchaya Mangpo Phothilimthana; Mohsen Lesani; Praveen Narayanan; Karthik Srinivasa Murthy; Rastislav Bodík; Amit Sabne; Charith Mendis (Proc. ACM Program. Lang., 2025)