team-ocean/veros

解決的問題

Veros 是一個高性能海洋模擬器，旨在取代海洋模擬中常見的舊有程式碼與複雜的設定流程。它讓研究人員能從簡單的理想化玩具模型，到高解析度的全球海洋模擬，無需在陳舊的程式環境中苦惱，即可順利執行。

運作方式

以純 Python 寫成的模擬器使用原始方程海洋模型。支援兩種後端：適用於小規模問題的 NumPy 後端，以及用於 CPU 與 GPU 加速的高效率 JAX 後端。系統透過 MPI 完全平行化，支援跨多個節點與多 GPU 架構的分散式執行。

適用對象

需要一個靈活、易於使用且可擴展的海洋模擬工具，能從筆電擴展至高效能運算叢集的海洋學家與氣候科學家。

主要特色

• 靈活的幾何結構：支援笛卡兒座標或偽球面座標下的完整交錯三維 C-網格幾何結構。
• 進階參數化：包含等中性混合、渦動能、湍流動能與內部波能量的參數化。
• 內建診斷功能：預先實作的能量通量、變數時間平均與垂直翻轉流函數診斷功能。
• 簡易設定：提供預先設定的理想化與真實情境設定，可透過簡單的 Python 程式碼直接執行。