Update equinox requirement from <=0.13.2,>=0.11.10 to >=0.11.10,<=0.13.3

[dependabot]

Updates the requirements on equinox to permit the latest version.

Release notes

Sourced from equinox's releases.

Equinox v0.13.3

• Performance: various module operations (setattr, initialisation, flatten, unflatten) should now be faster. In particular we now codegen flatten/unflatten functions. (Thanks @​nstarman! #1119, #1124, #1138, #1140)
• Feature: added EQX_ON_ERROR=off as a valid option. (Thanks @​michael-0brien! #1134)
• Compatibility:
  • JAX >=0.8.2 renamed jax.core.get_aval -> jax.typeof (#1157).
  • JAX >=0.8.2 deprecated jax.core.{subjaxprs, mapped_aval, unmapped_aval} (#1157).
  • JAX >=0.7.2 sometimes converts errors in jax.pure_callback in to ValueErrors; we now catch these appropriately (#1156, #1157).
• Misc: added support for pretty-printing eqx.Enumeration.
• Misc: tests now run on GPU (Thanks @​Artur-Galstyan! #1146)

Full Changelog: patrick-kidger/equinox@v0.13.2...v0.13.3

Commits

• 09e19a6 Bump to 0.13.3
• e6f9bd1 Added Enum pdoc
• 028b2a4 Make Static be Generic
• 235e3d0 Fix warnings from mapped_aval being made private
• 69ef176 Fixed errors after non-errors. (#1156)
• e6c07b4 Documentation fixes
• a47ffd0 EQX_ON_ERROR=off option allows user to switch off jax.lax.cond calls (#1134)
• d171883 Added link for opening a PR
• 2540dc3 Make tests pass on nvidia gpu (#1146)
• 0277a30 perf: code generation
• 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)

[github-actions]

Memory benchmark result

|               Test Name                |      %Δ      |    Master (MB)     |      PR (MB)       |    Δ (MB)    |    Time PR (s)     |  Time Master (s)   |
| -------------------------------------- | ------------ | ------------------ | ------------------ | ------------ | ------------------ | ------------------ |
  test_objective_jac_w7x                 |    5.22 %    |     3.880e+03      |     4.083e+03      |    202.56    |       38.86        |       37.37        |
  test_proximal_jac_w7x_with_eq_update   |   -0.07 %    |     6.586e+03      |     6.581e+03      |    -4.77     |       164.03       |       166.37       |
  test_proximal_freeb_jac                |   -0.01 %    |     1.319e+04      |     1.319e+04      |    -1.51     |       85.05        |       85.04        |
  test_proximal_freeb_jac_blocked        |    0.28 %    |     7.511e+03      |     7.532e+03      |    21.10     |       75.02        |       74.51        |
  test_proximal_freeb_jac_batched        |   -0.43 %    |     7.502e+03      |     7.470e+03      |    -32.60    |       74.02        |       74.32        |
  test_proximal_jac_ripple               |   -0.16 %    |     3.472e+03      |     3.466e+03      |    -5.54     |       66.10        |       66.50        |
  test_proximal_jac_ripple_bounce1d      |    2.34 %    |     3.548e+03      |     3.631e+03      |    82.95     |       77.25        |       78.48        |
  test_eq_solve                          |    0.67 %    |     2.022e+03      |     2.036e+03      |    13.63     |       95.31        |       94.70        |

For the memory plots, go to the summary of Memory Benchmarks workflow and download the artifact.

[dpanici]

Potentialy fixed by equinox 0.13.4: patrick-kidger/equinox#1172

[dependabot]

A newer version of equinox exists, but since this PR has been edited by someone other than Dependabot I haven't updated it. You'll get a PR for the updated version as normal once this PR is merged.