jax<=0.4.23,>=0.4.13
jaxlib<=0.4.23,>=0.4.13
scipy<1.13
optax>=0.1.4
tree-math>0.1.0
matplotlib
networkx
