pytorch-forecasting

PyTorch Forecasting 是一个基于 PyTorch 的时序预测专用库，专注于支持复杂时间序列场景下的深度学习建模。它集成 PyTorch Lightning 实现高效 GPU/CPU 训练与自动日志记录，内置 Temporal Fusion Transformer（TFT）、DeepAR、N-BEATS 等前沿模型，并提供标准化数据加载器、特征工程工具及不确定性估计能力。适用于具有多变量、协变量、不规则采样或长期依赖结构的业务时序预测任务，如销售预测、能源负荷预测、库存管理等工业与金融场景。