Equinox v0.11.8

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 #878

Full Changelog: v0.11.7...v0.11.8