MABSplit: Faster Forest Training Using Multi-Armed Bandits. - Mo Tiwari; Ryan Kang; Je-Yong Lee; Sebastian Thrun; Chris Piech; Ilan Shomorony; Martin Jinye Zhang (CoRR, 2022)

Faster Maximum Inner Product Search in High Dimensions. - Mo Tiwari; Ryan Kang; Je-Yong Lee; Luke Lee; Chris Piech; Sebastian Thrun; Ilan Shomorony; Martin Jinye Zhang (CoRR, 2022)

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