RTop-K: Ultra-Fast Row-Wise Top-K Selection for Neural Network Acceleration on GPUs.

Xi Xie, Yuebo Luo, Hongwu Peng, Caiwen Ding

Browse the full ICLR paper archive.