kuji 0.1.6

Stochastic sampling primitives: Gumbel-Softmax, reservoir sampling, and latent permutations
Documentation