research
∙
10/04/2021
A Review of the Gumbel-max Trick and its Extensions for Discrete Stochasticity in Machine Learning
The Gumbel-max trick is a method to draw a sample from a categorical dis...
share
research
∙
09/24/2021
Unbiased Gradient Estimation with Balanced Assignments for Mixtures of Experts
Training large-scale mixture of experts models efficiently on modern har...
research
∙
02/23/2021
Deep Policy Dynamic Programming for Vehicle Routing Problems
Routing problems are a class of combinatorial problems with many practic...
research
∙
02/14/2020
Estimating Gradients for Discrete Random Variables by Sampling without Replacement
We derive an unbiased estimator for expectations over discrete random va...
research
∙
03/14/2019