Dinosaur:用于全球大气建模的可微分动力学
作者:Jamie A. Smith, Dmitrii Kochkov, Peter Norgaard, Janni Yuval, Stephan Hoyer
Dinosaur是一个用JAX编写的全球大气建模谱系动力核心:
- 动力学:Dinosaur求解浅水方程,以及sigma坐标系下的原始方程(湿和干)。
- 自动微分:Dinosaur支持JAX中的正向和反向自动微分。
- 加速:Dinosaur设计为在现代加速硬件(GPU/TPU)上高效运行,包括跨多个设备的并行化。
更多详情,请参阅我们关于神经环流模型的论文。
使用说明
Dinosaur是一个实验性研究项目,我们仍在努力完善文档。
目前我们有复现动力核心两个标准测试案例的示例:
我们建议使用Google Colab的GPU运行时来运行它们。
你也可以在本地安装Dinosaur:pip install git+https://github.com/google-research/dinosaur
相关项目
如果你喜欢Dinosaur,你可能也会喜欢SpeedyWeather.jl,它用Julia解决类似的方程。
贡献
详情请参见CONTRIBUTING.md
。我们欢迎用户贡献,但在开始重要工作之前,请联系我们(通过GitHub或电子邮件)进行协调。
许可证
Apache 2.0;详情请参见LICENSE
。