Awesome

ai-models-fuxi

ai-models-fuxi is an ai-models plugin to run Fudan's FuXi.

FuXi: A cascade machine learning forecasting system for 15-day global weather forecast,arXiv preprint: 2306.12873, 2022. https://arxiv.org/pdf/2306.12873.pdf

FuXi was created by Lei Chen, Xiaohui Zhong, Feng Zhang, Yuan Cheng, Yinghui Xu, Yuan Qi, Hao Li. It is released by Fudan University.

The model weights are made available under the terms of the BY-NC-SA 4.0 license. The commercial use of these models is forbidden. Please download the pre-trained models from Google drive: https://drive.google.com/drive/folders/1CXxv-s4DdIHVxSCGsDLMn38n8Shk3fRs

For use other input data eg: ERA5 and GFS see https://github.com/tpys/FuXi.git for further details.

Installation

To install the package, run:

pip install ai-models-fuxi

This will install the package and its dependencies, in particular the ONNX runtime. The installation script will attempt to guess which runtime to install. You can force a given runtime by specifying the the ONNXRUNTIME variable, e.g.:

ONNXRUNTIME=onnxruntime-gpu pip install ai-models-fuxi