v12.0.1

danielward27 released this 13 Mar 21:00

· 153 commits to main since this release

0947b63

Change jax.tree -> jax.tree_util to maintain compatibility with earlier jax versions. Thanks @mdmould!

What's Changed

Replace jax.tree API with jax.tree_util for backwards compatibility by @mdmould in #143
Test for weight normalization by @danielward27 in #144
Version bump by @danielward27 in #145

Full Changelog: v12.0.0...v12.0.1

Contributors

danielward27 and mdmould

Assets 2