Inox

The ongoing march of small tweaks progresses.

Main changes this release:
- `eqx.{is_array,is_inexact_array}` now return `True` for `np.ndarray`s rather than `False`. This is technically a breaking change, hence the new minor version bump. Rationale in 202.
- We now use [`jaxtyping`](https://github.com/google/jaxtyping). Hurrah!

Other changes:
* make sequential module immutable by jenkspt in https://github.com/patrick-kidger/equinox/pull/195
* Add support for asymmetric padding in Conv and ConvTransposed. by Gurvan in https://github.com/patrick-kidger/equinox/pull/197

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

**Full Changelog**: https://github.com/patrick-kidger/equinox/compare/v0.7.1...v0.8.0

Autogenerated release notes as follows:

What's Changed
* Fixed `NotImplementedError` when computing gradients of `stateful` models by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/191
* fix attention with mask and add tests by uuirs in https://github.com/patrick-kidger/equinox/pull/190

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

**Full Changelog**: https://github.com/patrick-kidger/equinox/compare/v0.7.0...v0.7.1

- Multiple bugfixes for differentiating through, and serialising, `eqx.experimental.BatchNorm`.
- This is the reason for the version bump: if you are using `eqx.experimental.{BatchNorm,SpectralNorm,StateIndex}` then the serialisation format has changed.
- Feature: `use_ceil` added to all pooling layers.

---

Autogenerated release notes as follows:

What's Changed
* Add len and iter methods to nn.Sequential by jenkspt in https://github.com/patrick-kidger/equinox/pull/174
* Add attention functions and tests by jenkspt in https://github.com/patrick-kidger/equinox/pull/181
* Fixed BatchNorm not de/serialising correctly by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/172
* Ordered tree map by paganpasta in https://github.com/patrick-kidger/equinox/pull/170
* added use_ceil to pooling by paganpasta in https://github.com/patrick-kidger/equinox/pull/176
* Dev by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/184


**Full Changelog**: https://github.com/patrick-kidger/equinox/compare/v0.6.0...v0.7.0

- Refactor: the serialisation format for `eqx.experimental.{BatchNorm,SpectralNorm,StateIndex}` under `eqx.tree_{de,}serialise_leaves` has been tweaked slightly to avoid an edge-case crash. [This is the reason for the minor version bump to 0.6.0, as this is technically a (very minor) compatibility break.]
- Refactor: changed from `jax.tree_map` to `jax.tree_util.tree_map` to remove all the deprecation warnings JAX has started giving.
- Feature: added `eqx.nn.Lambda` (for use with `eqx.nn.Sequential`)
- Feature: added `eqx.default_{de,}serialise_filter_spec` (for use `eqx.tree_{de,}serialise_leaves).
- Bugfix: fixed `BatchNorm` crashing under `jax.grad`.
- Documentation: lots of tidy-ups and improvements.

---

Autogenerated release notes as follows:

What's Changed
* Doc tweak by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/141
* Fix GroupNorm channels argument and docstring by jenkspt in https://github.com/patrick-kidger/equinox/pull/148
* make `Sequential` indexable and add tests by jenkspt in https://github.com/patrick-kidger/equinox/pull/153
* replace tree_* with tree_util.tree_* to avoid jax warning messages by amir-saadat in https://github.com/patrick-kidger/equinox/pull/156
* Extend deserial filter by paganpasta in https://github.com/patrick-kidger/equinox/pull/145
* added lambda_layer to composites by paganpasta in https://github.com/patrick-kidger/equinox/pull/158
* Tweaked docs for Lambda by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/159
* Tweaked intro docs to improve readability of filtering by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/160
* Batch norm grad crash fix by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/162
* added outputs to the StateIndex example by paganpasta in https://github.com/patrick-kidger/equinox/pull/164
* Fixed crash when serialising StateIndices without saved state by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/167
* v0.6.0 by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/169

New Contributors
* jenkspt made their first contribution in https://github.com/patrick-kidger/equinox/pull/148
* amir-saadat made their first contribution in https://github.com/patrick-kidger/equinox/pull/156

**Full Changelog**: https://github.com/patrick-kidger/equinox/compare/v0.5.6...v0.6.0

Autogenerated release notes as follows:

What's Changed
* Adaptive avg pool 1d by paganpasta in https://github.com/patrick-kidger/equinox/pull/129
* `{Avg,Max}Pool{1,2,3}D` -> `{Avg,Max}Pool{1,2,3}d`. Removed wrong stride default. by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/135
* Tweaked AdaptivePool by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/139
* Adds adaptive pooling by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/140

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

**Full Changelog**: https://github.com/patrick-kidger/equinox/compare/v0.5.5...v0.5.6

Autogenerated release notes as follows:

What's Changed
* Fix doc typo by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/130
* Updated pooling docs with init and call by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/131
* Doc fix by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/132
* Tidied helper into a relative import by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/133
* minor bug fix by patrick-kidger in https://github.com/patrick-kidger/equinox/pull/134


**Full Changelog**: https://github.com/patrick-kidger/equinox/compare/v0.5.4...v0.5.5