search

arnauqb / GumbelSoftmax.jl

Julia implementation of the Gumbel-Softmax reparametrization trick compatible with Zygote and ForwardDiff
MIT License
6 stars 0 forks source link

Forwarddiff #3

Closed arnauqb closed 1 year ago

arnauqb commented 1 year ago