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

Solve bug where Random was not being imported for CUDA calculation #2

Closed arnauqb closed 1 year ago