pytest>=7
jax>=0.4.26
numpy>=1.24
nvtx

[plot]
matplotlib>=3.6
graphviz>=0.20
