Skip to content

Fix autoregressive sampling

Vicentini Filippo requested to merge github/fork/wdphy16/autoreg_fix into master

Created by: wdphy16

  • Reimplement batch_choice without scan, so it's faster a lot
  • Change n_chains to n_chains_per_rank in ARDirectSampler
  • Use logsumexp in the last layer of ARNN*, and remove ARNN*.eps

Changes to the public API:

  • ARNN*.eps is removed.

Merge request reports