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.
To install the package, run:
pip install ai-models-fuxiThis 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