SIMPLE: A Gradient Estimator for $k$-subset Sampling

Code for paper at: arxiv.org/abs/2210.01941