kuji 0.1.4

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