Review for NeurIPS paper: Differentiable Top-k with Optimal Transport