Inox

This is primarily a compatibility release.

- Fixes for compatibility with JAX 0.5.1 (959, 960).
- Fixes for compatibility with pyright 1.1.394 (956, 960).
- `eqx.nn.Linear(0, ...)` no longer crashes runs (propagating zero-size tensors). (Thanks aseyboldt! 950)
- Pretty-printing (`eqx.tree_pformat` and `eqx.tree_pprint`) now uses the new [Wadler-Lindig](https://github.com/patrick-kidger/wadler_lindig) pretty-printing library. (#924)
- Many doc improvements (Thanks matthewfeickert TugdualKerjan struan-robertson danielward27! 930, 937, 941, 950)

New Contributors
* matthewfeickert made their first contribution in https://github.com/patrick-kidger/equinox/pull/930
* struan-robertson made their first contribution in https://github.com/patrick-kidger/equinox/pull/941
* aseyboldt made their first contribution in https://github.com/patrick-kidger/equinox/pull/950
* TugdualKerjan made their first contribution in https://github.com/patrick-kidger/equinox/pull/947

**Full Changelog**: https://github.com/patrick-kidger/equinox/compare/v0.11.11...v0.11.12

JAX 0.4.38 moved a number of APIs with a deprecation warning, e.g. `jax.core.Jaxpr -> jax.extend.core.Jaxpr`. With this release we've updated and are back to being warning-free under this JAX release! (Thanks FFroehlich DrJessop, 913, 915, 917)

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

**Full Changelog**: https://github.com/patrick-kidger/equinox/compare/v0.11.10...v0.11.11

This is a JAX 0.4.36 compatibility release.

With this release, JAX changed how custom primitive rules are called (they are always called, instead of only when the data requires them to be). That requires some updates in Equinox to avoid crashes in the downstream ecosystem. (https://github.com/patrick-kidger/diffrax/issues/532, https://github.com/jax-ml/jax/issues/25289 + links therein.)


**Full Changelog**: https://github.com/patrick-kidger/equinox/compare/v0.11.9...v0.11.10

This is a (important) bugfix release.

* Fix filter_vmap with out_axes!=0,1 producing outputs with the wrong axis order. (Thanks remifan! 900, 901)

**Full Changelog**: https://github.com/patrick-kidger/equinox/compare/v0.11.8...v0.11.9

The main thing for this release is JAX 0.4.34 compatibility -- JAX introduced breaking changes in this release that we are now compatible with. (871)

Bugfixes

* Accessing the concrete implementation of an abstract class attribute within `__init_subclass__` should no longer crash. (Plus probably better-behaved `__init_subclass__` overall.)

Miscellaneous

* JAX 0.4.33 introduced a change that broke `eqx.error_if`'s nice displaying of error message. With this release then we are back to having nice error messages again!
* `eqx.nn.StateIndex` can now be passed through `jax.jit` (and not just `eqx.filter_jit`). (Thanks NeilGirdhar! 843)
* Normalization layers now upcast to at least 32-bit precision. (Thanks AakashKumarNain! 876)
* Poetry has a bug in its interpretation of `~=` version constraints. We now work around that for better compatibility with certain kinds of Poetry installations. (Thanks norpadon! 878)

Documentation

* Updated CNN example to work with recent JAX versions. (Thanks pasq-cat! 880, 881)
* Update `eqx.tree_at` documentation for clarity. (Thanks jeertmans! 872, 874, 877)

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

**Full Changelog**: https://github.com/patrick-kidger/equinox/compare/v0.11.7...v0.11.8

Quick release. JAX 0.4.32 / 0.4.33 just introduced a breaking change; this release ensures Equinox is compatible with this. (856)


**Full Changelog**: https://github.com/patrick-kidger/equinox/compare/v0.11.6...v0.11.7