Deep Graph Infomax
Automatic differentiation in PyTorch