Skip to content

v0.3.0

Latest

Choose a tag to compare

@LegrandNico LegrandNico released this 16 Jun 10:55
772f21d

This is a major update with breaking changes that introduce both new models and a new API.

  • We introduce new "relaxed" updates that drop the mean-field assumption in the posterior and prediction update, resulting in uncertainties flowing both top-down and bottom-up via value coupling. This is of particular interest for deep networks.
  • The DeepNetwork class now uses Equinox internally and scans over the ayer axis, allowing to train deeper model without hitting the compilation burden.
  • All main classes let the user control volatility updates, mean-field updates, and clipping threshold. update_type is now volatility_updates.

What's Changed

Full Changelog: v0.2.12...v0.3.0