graphviz>=0.14.1
numpy
IPython
matplotlib

[all]
tensorflow
torch

[tensorflow]
tensorflow

[torch]
torch
