Shypd.aiThe largest open index of AI toolsPyTorch Geometric Temporal is an open-source library for spatiotemporal signal processing with neural machine learning models. It extends PyTorch Geometric for dynamic and temporal graphs, offering various deep learning methods and an easy-to-use dataset loader.
Trending
About
PyTorch Geometric Temporal is an open-source Python library designed for spatiotemporal signal processing using neural machine learning models. Built as a temporal extension for PyTorch Geometric, it provides a comprehensive collection of dynamic and temporal geometric deep learning, embedding, and spatio-temporal regression methods from published research. The library features an easy-to-use dataset loader, train-test splitter, and temporal snapshot iterator for dynamic and temporal graphs, with natural GPU support. It includes benchmark datasets across domains like epidemiological forecasting, sharing economy, energy production, and web traffic management. Key advancements include index-batching for improved spatiotemporal memory efficiency and support for distributed data parallel training with Dask-DDP, making it scalable for larger datasets. It also integrates well with PyTorch Lightning for flexible training environments.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
