torch
torchvision
torchtext

graphviz