Spartan: Differentiable Sparsity via Regularized Transportation - Tai, Kai Sheng; Tian, Taipeng; Lim, Ser Nam (Advances in Neural Information Processing Systems, 2022)