distrax icon indicating copy to clipboard operation
distrax copied to clipboard

Published 20 hours ago • verified publisher icon google-deepmind

Feature request: RelaxedOneHotCategorical distribution

Open lkhphuc opened this issue 2 years ago • 1 comments

TFD: https://www.tensorflow.org/probability/oryx/api_docs/python/oryx/distributions/RelaxedOneHotCategorical

lkhphuc avatar Mar 12 '22 03:03 lkhphuc

Is there still interest in a gumbel softmax?

wbrenton avatar Dec 21 '22 01:12 wbrenton