SalesforceAIResearch/uni2ts
A PyTorch library for pre-training, fine-tuning, and evaluating universal time series transformer models (Moirai, Moirai-MoE) at scale.

Velocity · 7d
+1.8
★ / day
Trend
→steady
star history
Uni2TS provides a unified framework for large-scale pre-training, fine-tuning, inference, and evaluation of time series forecasting transformers. It implements the Moirai and Moirai-MoE model families and includes the GIFT-Eval benchmark for general time series forecasting evaluation. The library offers model weights on Hugging Face with a leaderboard for comparing forecasting performance.