jax[cpu]
compas<2.0
scipy
equinox
jaxopt
