Skip to content

Equinox v0.3.0

Choose a tag to compare

@github-actions github-actions released this 27 Mar 20:28
· 910 commits to main since this release
2dba6cf

Three main things in this release.

  1. equinox.experimental.BatchNorm. Hurrah, that's nice to have.
  2. Very interesting from a technical point of view: stateful operations. In this case, equinox.experimental.{get_state, set_state, StateIndex}. These are the technology used to update the statistics of BatchNorm without requiring the user to faff around outputting the model themselves. They work by wrapping jax.experimental.host_callback.call to save and load external state on demand. Which is pretty magic, so these should really be used sparingly...
  3. Removed several old pieces of deprecated functionality: equinox.jitf and so on.

Autogenerated release notes as follows:

What's Changed

Full Changelog: v0.2.2...v0.3.0