pysb dimerization test fails with JAX

[dweindl]

python/tests/test_jax.py::test_dimerization fails with:

E               AssertionError: 
E               Not equal to tolerance rtol=1e-05, atol=1e-05
E               field x does not match
E               +inf location mismatch:
E                ACTUAL: array([[0.000000e+00, 0.000000e+00, 0.000000e+00],
E                      [4.223384e-02, 5.399623e-01, 4.494823e-04],
E                      [         inf,          inf,          inf],...
E                DESIRED: array([[0.000000e+00, 0.000000e+00, 0.000000e+00],
E                      [4.223384e-02, 5.399623e-01, 4.494823e-04],
E                      [6.493001e-02, 1.048842e+00, 2.849623e-03],...

python/tests/test_jax.py:275: AssertionError

https://github.com/AMICI-dev/AMICI/actions/runs/22293873991/job/64486499503

[FFroehlich]

python/tests/test_jax.py::test_dimerization fails with:

E               AssertionError: 
E               Not equal to tolerance rtol=1e-05, atol=1e-05
E               field x does not match
E               +inf location mismatch:
E                ACTUAL: array([[0.000000e+00, 0.000000e+00, 0.000000e+00],
E                      [4.223384e-02, 5.399623e-01, 4.494823e-04],
E                      [         inf,          inf,          inf],...
E                DESIRED: array([[0.000000e+00, 0.000000e+00, 0.000000e+00],
E                      [4.223384e-02, 5.399623e-01, 4.494823e-04],
E                      [6.493001e-02, 1.048842e+00, 2.849623e-03],...

python/tests/test_jax.py:275: AssertionError

https://github.com/AMICI-dev/AMICI/actions/runs/22293873991/job/64486499503

@BSnelling is already having a look at this, probably due to diffrax/optimistix release

[AlwaysTwilight]

Agent run 95dfdbe8fdae483680fc1fd1195d57cc started.

[AlwaysTwilight]

Agent run 95dfdbe8fdae483680fc1fd1195d57cc failed: Command '['docker', 'run', '--rm', '--cpus', '1.0', '--memory', '2g', '-v', 'D:\SWE_Agents\.work\95dfdbe8fdae483680fc1fd1195d57cc:/work', '-w', '/work', 'python:3.11-slim', 'sh', '-lc', 'if ! command -v git >/dev/null 2>&1; then apt-get update -y >/dev/null 2>&1 && apt-get install -y git >/dev/null 2>&1; fi; git clone https://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@github.com/AMICI-dev/AMICI.git repo']' timed out after 600 seconds