jax>=0.4.13
jaxlib>=0.4.13
scipy
optax>=0.1.4
tree-math>0.1.0
matplotlib
networkx

[nir]
nir>1.0.7
