Bump equinox from 0.11.3 to 0.11.6

[dependabot]

Bumps equinox from 0.11.3 to 0.11.6.

Release notes

Sourced from equinox's releases.

Equinox v0.11.6

This is primarily a bug fix release.

• Runtime error messages (those from eqx.error_if, in particular when wrapped with eqx.filter_jit) should now be compatible with PyCharm's debugger, and with certain multithreaded contexts. (Thanks @​adam-hartshorne, @​dlwh! #828, #844, #849)

• Marking a jax.Array or np.ndarray as an eqx.field(static=True) will now raise a warning. This was technically okay as long as you use it in certain very narrow contexts (e.g. to smuggle it into a JIT'd region without being traced), but in practice it was nearly always just a common new-user footgun. (Thanks @​lockwo! #800)

• Using eqx.tree_at for replacing empty tuples is improved. (Thanks @​danielward27! #818, #819)

• eqx.nn.RotaryEmbedding no longer promote input dtypes to at least float32. (Thanks @​knyazer! #836)

• Mypy now understands that eqx.Modules are dataclasses. (Pyright always did, but mypy needed a slightly different approach to appreciate this fact.) (Thanks @​NeilGirdhar! #822)

• Multiple eqx.Modules participating in co-operative multiple inheritance (at least 5 inheriting from each other seem to be necessary?), with some of them overriding the __post_init__s of others, should now follow their expected resolution order. (Thanks @​NeilGirdhar! #832, #834)

• We now have a .editorconfig file, (thanks @​NeilGirdhar! #821)

• Doc improvements. (Thanks @​garymm, @​ColCarroll! #804, #805)

New Contributors

• @​garymm made their first contribution in patrick-kidger/equinox#804
• @​ColCarroll made their first contribution in patrick-kidger/equinox#805
• @​NeilGirdhar made their first contribution in patrick-kidger/equinox#823

Full Changelog: patrick-kidger/equinox@v0.11.5...v0.11.6

Equinox v0.11.5

JAX compatibility

Recent versions of JAX (0.4.28+) have made some changes to:

• Hashing of tracers;
• Tree-map'ing over Nones;
• Callbacks;
• Pretty-printing.

With this update, we should now be compatible with both old and new versions of JAX: this fixes both some new crashes, and some new warnings. (#719, #724, #753, #758, thanks @​jakevdp, @​hawkinsp!)

Better errors

• The error messages from eqx.error_if are now substantially more informative: they include traceback information including the stack, and mention the availability of the EQX_ON_ERROR variable. We also do a much better job hiding the large unhelpful printouts that XLA gives by default. (#785, #803)

• The default value of EQX_ON_ERROR_BREAKPOINT_FRAMES is now 1. (#777) The impact of this is that using eqx.error_if alongside EQX_ON_ERROR=breakpoint will now:

  • reliably always open a debugger, rather than sometimes crashing at trace-time due to upstream JAX bug #16732.
  • however, by default the debugger will no longer include any additional stack frames above it (accessed via u).
  • much of the above is now explained in a printed-out informative message prior to the debugger opening.

Bugfixes

• eqx.filter_{jacfwd, jacrev} now only apply filtering to their inputs but not their outputs. Previously this was problematic as there was no way to represent static-input-by-static-output in the returned Jacobian, so pieces were silently dropped. (#734, thanks @​lockwo!)

... (truncated)

Commits

• 05b01c1 Maybe fix doc build?
• ea1c1cc Revert "Add [test] and update CONTRIBUTING (#848)"
• 71f372c Version bump
• 64411ed Improved closure-conversion error.
• ec67b51 Update upload-artifact@v4
• 5f588e5 Improve state error
• ec4747e Filtering out jax.pure_callback failures via logging; added test.
• ffc0985 fix: RoPE embeddings do not preserve dtypes (#836)
• 5a40d2e Add [test] and update CONTRIBUTING (#848)
• 97ac55a build: bump up precommit hooks (#837)
• Additional commits viewable in compare view

Dependabot will resolve any conflicts with this PR as long as you don't alter it yourself. You can also trigger a rebase manually by commenting @dependabot rebase.

---

Dependabot commands and options

You can trigger Dependabot actions by commenting on this PR:

• @dependabot rebase will rebase this PR
• @dependabot recreate will recreate this PR, overwriting any edits that have been made to it
• @dependabot merge will merge this PR after your CI passes on it
• @dependabot squash and merge will squash and merge this PR after your CI passes on it
• @dependabot cancel merge will cancel a previously requested merge and block automerging
• @dependabot reopen will reopen this PR if it is closed
• @dependabot close will close this PR and stop Dependabot recreating it. You can achieve the same result by closing it manually
• @dependabot show <dependency name> ignore conditions will show all of the ignore conditions of the specified dependency
• @dependabot ignore this major version will close this PR and stop Dependabot creating any more for this major version (unless you reopen the PR or upgrade to it yourself)
• @dependabot ignore this minor version will close this PR and stop Dependabot creating any more for this minor version (unless you reopen the PR or upgrade to it yourself)
• @dependabot ignore this dependency will close this PR and stop Dependabot creating any more for this dependency (unless you reopen the PR or upgrade to it yourself)