TPU-KNN: K Nearest Neighbor Search at Peak FLOP/s. - Felix Chern; Blake Hechtman; Andy Davis; Ruiqi Guo; David Majnemer; Sanjiv Kumar (NeurIPS, 2022)

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)

eXmY: A Data Type and Technique for Arbitrary Bit Precision Quantization. - Aditya Agrawal; Matthew Hedlund; Blake Hechtman (CoRR, 2024)