Dinosaur：全局大气建模的不同iable动力学

作者：Jamie A. Smith，Dmitrii Kochkov，Peter Norgaard，Janni Yuval，Stephan Hoyer

Dinosaur是用JAX编写的全局大气建模的谱动力学核心

• 动力学：Dinosaur在sigma坐标上求解浅水方程和原始方程（湿和干）。
• 自动微分：Dinosaur支持JAX中的前向和后向自动微分。
• 加速：Dinosaur设计用于在现代加速硬件（GPU/TPU）上高效运行，包括跨多个设备的并行化。

更多详情，请参阅我们关于神经一般环流模型的论文。

使用说明

Dinosaur是一个仍在努力文档化的实验性研究项目。

我们目前有一些示例，复制了动力学核心的两个标准测试用例

• 巴洛克不稳定性
• Held-Suarez强迫

我们建议使用带有GPU运行时的Google Colab运行它们。您也可以本地安装Dinosaur： pip install git+https://github.com/google-research/dinosaur

参见

如果您喜欢Dinosaur，您可能也会喜欢SpeedyWeather.jl，它在Julia中求解类似方程。

贡献

请参阅CONTRIBUTING.md以获取详细信息。我们欢迎用户贡献，但在开始重大工作之前，请先联系（GitHub或电子邮件）以协调。

许可证

Apache 2.0；请参阅LICENSE以获取详细信息。