Differentiable Top-k Operator with Optimal Transport | Read Paper on Bytez