wouterkool/ancestral-gumbel-top-k-sampling

Ancestral Gumbel-Top-k Sampling

Jupyter NotebookMIT