Update dependencies manually

[YigitElma]

Updating the dependencies manually before the new release. The Dependabot issue persists.

Question:
Looks like jax.tree_util.tree_broadcast was added with jax 0.6.2. Should I try to copy their implementation to backend.py or should I just update min jax version? Affect jax version are 0.5.0, 0.5.3 and 0.6.1 (we don't allow others anyway)

[github-actions]

Memory benchmark result

|               Test Name                |      %Δ      |    Master (MB)     |      PR (MB)       |    Δ (MB)    |    Time PR (s)     |  Time Master (s)   |
| -------------------------------------- | ------------ | ------------------ | ------------------ | ------------ | ------------------ | ------------------ |
  test_objective_jac_w7x                 |    0.55 %    |     4.141e+03      |     4.163e+03      |    22.77     |       32.27        |       29.75        |
  test_proximal_jac_w7x_with_eq_update   |    0.16 %    |     6.836e+03      |     6.847e+03      |    11.16     |       153.86       |       151.21       |
  test_proximal_freeb_jac                |   -0.18 %    |     1.342e+04      |     1.340e+04      |    -23.57    |       80.22        |       80.92        |
  test_proximal_freeb_jac_blocked        |   -0.69 %    |     7.792e+03      |     7.739e+03      |    -53.88    |       70.69        |       72.71        |
  test_proximal_freeb_jac_batched        |    0.70 %    |     7.710e+03      |     7.764e+03      |    53.75     |       70.78        |       71.50        |
  test_proximal_jac_ripple               |    1.83 %    |     3.550e+03      |     3.615e+03      |    64.99     |       55.55        |       55.34        |
  test_proximal_jac_ripple_bounce1d      |   -0.10 %    |     3.834e+03      |     3.830e+03      |    -3.80     |       73.16        |       68.12        |
  test_eq_solve                          |    1.04 %    |     2.080e+03      |     2.102e+03      |    21.66     |       88.54        |       88.53        |

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

[YigitElma]

@f0uriest what do you think about bumping the minimum jax version to 0.6.2? It is almost a year old now (June 17, 2025).

[YigitElma]

• Try test_solve_bounds with svd
• Bump the min version of jax