Inox

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 `RuntimeError`s 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.Module`s) 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 https://github.com/patrick-kidger/equinox/pull/58
* Remove default bias in `MultiheadAttention` by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/60
* Improvements to stateful by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/61
* Improved pretty-printing by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/62
* Bump version by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/63


**Full Changelog**: https://github.com/patrick-kidger/equinox/compare/v0.3.2...v0.4.0

Autogenerated release notes as follows:

What's Changed
* Version 0.3.2: SpectralNorm and more! by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/55


**Full Changelog**: https://github.com/patrick-kidger/equinox/compare/v0.3.1...v0.3.2

Autogenerated release notes as follows:

What's Changed
* Fix function names when logging compilation progress by marcelroed in https://github.com/patrick-kidger/equinox/pull/52

New Contributors
* marcelroed made their first contribution in https://github.com/patrick-kidger/equinox/pull/52

**Full Changelog**: https://github.com/patrick-kidger/equinox/compare/v0.3.0...v0.3.1

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
* Score based example by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/42
* minor doc fixes by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/43
* Version 0.3.0 -- BatchNorm and stateful by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/50


**Full Changelog**: https://github.com/patrick-kidger/equinox/compare/v0.2.2...v0.3.0

Added several new layers:
- LayerNorm
- MultiheadAttention
- ConvTranspose, ConvTranspose1d, ConvTranspose2d, ConvTranspose3d
- Embedding

(Thanks to andyehrenberg for implementing much of this, and to lucidrains for reviewing the implementation of attention.)

Autogenerated release notes as follows:

What's Changed
* ConvTranspose layers, MultiheadAttention, lookup embeddings, LayerNorm by andyehrenberg in https://github.com/patrick-kidger/equinox/pull/34
* Tidied; simplified; generalised ConvTranspose implementation. by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/40
* Attention, Transposed Convolutions, Embeddings, LayerNorm by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/38


**Full Changelog**: https://github.com/patrick-kidger/equinox/compare/v0.2.1...v0.2.2

Autogenerated release notes as follows:

What's Changed
* Added automated releases. by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/39
* Added pretty-printing for Modules by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/41


**Full Changelog**: https://github.com/patrick-kidger/equinox/compare/v0.2.0...v0.2.1