Relevant links:
Temporal Graph Networks paper --> https://arxiv.org/pdf/2006.10637.pdf
TGN source code --> https://pytorch-geometric.readthedocs.io/en/latest/_modules/torch_geometric/nn/models/tgn.html?highlight=Temporal
Example using tgn with torch_geometric --> https://github.com/rusty1s/pytorch_geometric/blob/master/examples/tgn.py
Overleaf with the description of the algorithm --> https://www.overleaf.com/project/604e45c9f214fb2c08748a58