pytorch-lightning
torch-geometric-temporal
