torch>=2.6.0
matplotlib
numpy
graphviz
