One of the authors here. I've read the paper. Brilliant work, especially the slicing implementation for denser token masks.