FlashTP: Fused, Sparsity-Aware Tensor Product for Machine Learning Interatomic Potentials.

Seung Yul Lee, Hojoon Kim, Yutack Park, Dawoon Jeong, Seungwu Han, Yeonhong Park, Jae W. Lee

Browse the full ICML paper archive.