@inproceedings{unbiasedgradientestimationwithbalanced,
  title = {Unbiased Gradient Estimation with Balanced Assignments for Mixtures of Experts},
  author = {Wouter Kool and Chris J. Maddison and andriy mnih},
  year = {2021},
  booktitle = {NeurIPS Workshop ICBINB 2021 12},
  eprint = {2109.11817},
  archivePrefix = {arXiv},
  url = {https://arxiv.org/abs/2109.11817v2},
}