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

[nir]
nir>=1.0.8

[test]
pytest

[test-nir]
pytest
nir>=1.0.8
