Test execution gets stuck in CI

[jobrachem]

Our test execution in the github CI currently gets stuck and runs into the 10 minute timeout. I suspect that the tests get stuck in the tests for the liesel.goose.optim module. The issue is new and did not show up during the latest PR with updates to the optim module: #196 (all checks passed).

Locally on my machine, the tests run quickly as usual; the doctests run in ~16 seconds and the unit tests run in less than 150 seconds.

The optax version is 0.2.3 in both cases, so that does not seem to be the problem.

[jobrachem]

See this test run:

• new: https://github.com/liesel-devs/liesel/actions/runs/10809444812/attempts/2
• old: https://github.com/liesel-devs/liesel/actions/runs/10809444812/attempts/1

Same commit, I just re-ran the test action. It previously worked, now it times out, seemingly stuck in the tests for optim.py.