jax<0.12,>=0.10
networkx<4,>=3.2.1
