torch-graph Graph package for Torch Requisites You might want to install Graphviz to benefit from the vizualisation functions.