jax[cpu]
compas<2.0
scipy
jaxopt
equinox
lineax
optimistix
