fastprogress>=0.2.0
jax>=0.3.13
jaxlib>=0.3.10
jaxopt>=0.5.5
