Welcome to TGLite’s documentation!¶
TGLite is a lightweight framework that provides core abstractions and building blocks for practitioners and researchers to implement efficient TGNN models. TGNNs, or Temporal Graph Neural Networks, learn node embeddings for graphs that dynamically change over time by jointly aggregating structural and temporal information from neighboring nodes.
TGLite employs an abstraction called a TBlock to represent the temporal graph dependencies when aggregating from neighbors, with explicit support for capturing temporal details like edge timestamps, as well as composable operators and optimizations. Compared to prior art, TGLite can outperform the TGL framework by up to 3x in terms of training time.
End-to-end training epoch time comparison on an Nvidia A100 GPU¶
Install TGLite¶
See Getting started for instructions on how to install the TGLite binaries. To install from source or for local development, refer to Building from source and Development mode.
Tutorials¶
We provide a set of tutorials to help you get started with TGLite. These tutorials cover the basics of using TGLite, as well as more advanced topics.
Quickstart: A step-by-step guide to train a TGNN model using TGLite.
How does TBlock work?: A tutorial on how to use the TBlock abstraction to implement TGNN models.
API
Note
This project is under active development.