Skip to content

0.3.1

Compare
Choose a tag to compare
@fritzo fritzo released this 13 Feb 19:00
· 977 commits to dev since this release

Dependency changes

  • Removes dependency on networkx.
  • Uses opt_einsum version 2.3.2.

New features

Minor changes and bug fixes

  • Renamed ubersum(..., batch_dims=...) (deprecated) to einsum(..., plates=...).
  • HMC - fix bug in initial trace setter and diagnostics, resolve slowness issues during adaptation, expose target_accept_prob and max_tree_depth as arguments to the constructor to allow finer grained control over hyper-parameters.
  • Many small fixes to the tutorials.