Equinox v0.4.0

A new minor release as there's a few minor breaking changes:

• Some of the projections in MultiheadAttention no longer have a bias by default (#60)
• equinox.experimental.{get_state,set_state,BatchNorm} now raises RuntimeErrors for many things; this is to match a change in how jax.experimental.host_callback raises errors in jaxlib>=0.3.5. (#63)

Besides this, a couple of more exciting (?) things:

• equinox.tree_pformat (which is used when printing equinox.Modules) now pretty-prints results much more neatly. (#62)
• equinox.experimental.{get_state,BatchNorm,SpectralNorm} are now substantially faster when run in inference mode. (#61)

Both of which sound pretty minor but both of which were technically really interesting to implement ;)

---

The pull requests in this release were:

• Linear doc tweaks by @patrick-kidger in #58
• Remove default bias in MultiheadAttention by @patrick-kidger in #60
• Improvements to stateful by @patrick-kidger in #61
• Improved pretty-printing by @patrick-kidger in #62
• Bump version by @patrick-kidger in #63

Full Changelog: v0.3.2...v0.4.0