Equinox v0.6.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 #141
• Fix GroupNorm channels argument and docstring by @jenkspt in #148
• make Sequential indexable and add tests by @jenkspt in #153
• replace tree_* with tree_util.tree_* to avoid jax warning messages by @amir-saadat in #156
• Extend deserial filter by @paganpasta in #145
• added lambda_layer to composites by @paganpasta in #158
• Tweaked docs for Lambda by @patrick-kidger in #159
• Tweaked intro docs to improve readability of filtering by @patrick-kidger in #160
• Batch norm grad crash fix by @patrick-kidger in #162
• added #outputs to the StateIndex example by @paganpasta in #164
• Fixed crash when serialising StateIndices without saved state by @patrick-kidger in #167
• v0.6.0 by @patrick-kidger in #169

New Contributors

• @jenkspt made their first contribution in #148
• @amir-saadat made their first contribution in #156

Full Changelog: v0.5.6...v0.6.0