NVIDIA выпустила открытую модель прогнозирования Kumo Forecast

NVIDIA представила Kumo Forecast — открытую модель многомерного прогнозирования временных рядов на базе трансформера MOMENT-1-large с 340 млн параметров. Код и веса доступны под лицензией Apache 2.0 на GitHub и Hugging Face, модель поддерживает retrieval по историческим данным и интерпретируемость прогнозов.

• Модель на базе MOMENT-1-large с энкодером на ~340 млн параметров
• Код и веса открыты под Apache 2.0 на GitHub и Hugging Face
• Поддерживает горизонты 256–2048 наблюдений и режим DARR с kNN-retrieval
• Работает на PyTorch с GPU Ampere и Hopper, тесты на A100 и H100