team-ocean/veros

解决的问题

Veros 是一个高性能海洋模拟器，旨在取代海洋建模中常见的老旧代码和复杂设置流程。它使研究人员能够从简单的理想化玩具模型运行到高分辨率全球海洋模拟，而无需在陈旧的编程环境中挣扎。

工作原理

用纯 Python 编写的模拟器使用原始方程海洋模型。它支持两种后端：适用于小规模问题的 NumPy 后端，以及用于 CPU 和 GPU 加速的高性能 JAX 后端。系统通过 MPI 完全并行化，支持在多个节点和多 GPU 架构上进行分布式执行。

适用人群

需要一个灵活、易用且可扩展的海洋建模工具，能够从笔记本电脑扩展到高性能计算集群的海洋学家和气候科学家。

主要亮点

• 灵活的几何结构：支持笛卡尔坐标或伪球面坐标下的完全交错三维 C-网格几何结构。
• 高级参数化：包含等中性混合、涡动能、湍流动能和内部波能量的参数化。
• 内置诊断功能：预实现的能量通量、变量时间平均和垂直翻转流函数诊断功能。
• 简易设置：提供预配置的理想化和真实场景设置，可通过简单的 Python 脚本直接运行。