jax>=0.3.4
equinox>=0.5.2
optax>=0.1.2
networkx>=2.6
