team-ocean/veros

The versatile ocean simulator, in pure Python, powered by JAX.

何を解決するか

Veros は、海洋モデルにおける伝統的なコードや複雑なセットアップワークフローを置き換えるために設計された高性能海洋シミュレータです。研究者が単純な理想化トイモデルから高解像度の全球海洋シミュレーションまで、古くさいプログラミング環境に苦労することなく実行できるようにします。

動作方法

純粋な Python で書かれたシミュレータは、原始方程式海洋モデルを使用しています。2つのバックエンドをサポートしています:小規模な問題向けの NumPy バックエンドと、CPU および GPU 加速向けの高性能 JAX バックエンド。システムは MPI を通じて完全に並列化されており、複数ノードおよびマルチ GPU アーキテクチャでの分散実行が可能になっています。

対象ユーザー

ラップトップからハイパフォーマンスコンピューティングクラスタまでスケーラブルな、柔軟でアクセスしやすく拡張可能な海洋モデルツールを必要とする海洋学者や気候科学者。

特徴

  • 柔軟な幾何学: カルテシアン座標または擬球面座標における完全にステガードな 3 次元 C-グリッド幾何学をサポート。
  • 高度なパラメータ化: 等中性混合、渦運動エネルギー、乱流運動エネルギー、内部波エネルギーのパラメータ化を含む。
  • 組み込み診断: エネルギーフラックス、変数時間平均、垂直循環ストリーム関数のための事前実装された診断機能。
  • 簡単なセットアップ: シンプルな Python スクリプトで実行可能な、事前構成された理想化および現実的な設定を提供。

関連

  • プロジェクト
  • プロジェクト
  • プロジェクト
  • プロジェクト
  • プロジェクト