jax>=0.4.25
jaxlib>=0.4.25
jraph
flax
optax

[all]
pymanopt>=2.0.1
