arnauqb/GumbelSoftmax.jl

Julia implementation of the Gumbel-Softmax reparametrization trick compatible with Zygote and ForwardDiff

语言：Julia · GitHub 星标：10 · Fork：1

在 GitHub 上查看原仓库