fastprogress>=1.0.0
jax>=0.4.12
jaxlib>=0.4.14
jaxopt>=0.8
optax>=0.1.7
typing-extensions>=4.4.0
