storchastic icon indicating copy to clipboard operation
storchastic copied to clipboard

Implement Gumbel-Rao Monte Carlo estimator

Open HEmile opened this issue 5 years ago • 0 comments

See https://openreview.net/pdf?id=Mk6PZtgAgfq Relatively straightforward to implement: Like a reparameterization with a modified (stochastic!) backward pass override. Overriding backward pass is already needed to implement normal straight-through Gumbel softmax, so just modify it.

HEmile avatar Oct 06 '20 15:10 HEmile