GitHub repository: radandreicristian/traffic
Primary language: Python
Description: An end-to-end traffic forecasting training pipeline and a collection of PyTorch traffic forecasting models that use graph neural PDEs and sub-quadratic attention.