AMReX-Astrophysics Microphysics: A set of microphysics routines for astrophysical simulation codes based on the AMReX library

AMReX-Astrophysics Microphysics に関する技術要約

問題提起
天体物理学のシミュレーションコードには、流体力学方程式を解くために、多様なマイクロフィジックス（微視的物理）の入力、具体的には反応ネットワーク、状態方程式（EOS）、輸送係数、および関連するソルバーが必要である。AMReX（Adaptive Mesh Refinement）エコシステム内には多くのコード（Castro、MAESTROeX、Quokka など）が存在するが、歴史的にそれぞれが独自のマイクロフィジックス実装を開発してきた。このような断片化は、開発努力の重複を招き、特にGPUオフローディングを利用するエクサスケール・アーキテクチャに向けた、これらのコードの協調的な最適化を妨げている。既存のマイクロフィジックス・ルーチンの集積は、現代のGPU要件に追いついていないか、あるいは特定の物理コンポーネント（例：EOSのみ）に限定されており、AMReXコミュニティ全体のための統一されたGPUネイティブのフレームワークを欠いている。

手法およびソフトウェア設計
AMReX-Astrophysics Microphysics ライブラリは、これらのルーチンを集約するために開発された。これは、レガシーなFortranベースのシステム（元は「Starkiller Microphysics」として知られていた）から、現代的なヘッダーオンリーのC++実装への移行を目的としている。設計思想は、以下の主要な技術戦略を通じて、GPUオフローディングとエクサスケール・ハードウェア上でのパフォーマンスを優先している。

• アーキテクチャと言語: 本ライブラリは完全にC++で記述されており、AMReXのデータ構造を活用することで、直接的なGPUメモリ割り当てと実行を容易にしている。物理ルーチンは単一のシミュレーションゾーンに対して動作するように設計されており、C++のラムダキャプチャを利用してゾーンをループ処理し、計算をGPUへオフロードする。
• パフォーマンスの最適化: GPUの効率を最大化するため、コードはアグレッシブなコンパイラ・インライン化を可能にするヘッダーオンリーの実装を採用している。また、C++17の if constexpr テンプレートを利用して不要な計算をコンパイル時に排除し、実行時のオーバーヘッドを削減している。
• ソルバーとネットワークの分離: コアとなる設計上の特徴は、反応ネットワークと積分器のデカップリング（分離）である。このモジュール性により、研究者は物理定義を書き換えることなく、異なる積分方法（例：硬いODE積分器であるVODE、ハイブリッド・パウエル法、またはRunge–K–Chebyshevなど）や結合モード（演算子分割法 vs スペクトル遅延補正法）を実験的に試すことができる。
• スレッド・ダイバージェンスの処理: 熱力学的状態がシミュレーションゾーン間で変化し、GPUカーネルにおけるスレッド・ダイバージェンス（分岐）を引き起こす可能性を認識し、ライブラリは積分ステップを制限する戦略を実装している。ゾーンごとに調整された許容誤差やヤコビアン近似を用いて積分の再試行を行うか、失敗をアプリケーションコードに返却する機能をサポートしている。
• 統合ツール: ライブラリには、カスタム反応ネットワークを生成するための pynucastro ツールや、ヤコビアンに必要な熱力学的微分を計算するためのC++ autodiff ライブラリが含まれている。古典的なFortranライブラリ（VODE、MINPACK、RKC）は、goto 文を排除して近代化し、アプリケーション固有のロジックを強化した上で、ヘッダーオンリーのC++実装へと変換されている。

主な貢献
本論文は、AMReX-Astrophysics Microphysics ライブラリを、AMReXコミュニティのための包括的かつ統一されたリソースとして提示している。その主な貢献は以下の通りである：

1. 統一された物理ライブラリ: 反応ネットワーク、EOS、熱伝導率、オパシティ、および核統計平衡ソルバーの共通セットを提供すること。
2. GPUネイティブの実装: これらのルーチンの完全なC++ベースのGPUオフロード版を提供し、現代的なアーキテクチャ上での高性能シミュレーションを可能にすること。
3. モジュール式ソルバー: 特定のシミュレーションのニーズ（例：爆発的流動におけるタイトな結合）に基づいて、入れ替えや組み合わせが可能な、硬いODE積分器および非線形システムソルバーの柔軟なスイートを提供すること。
4. コンパイル効率: 反応ネットワークの選択にコンパイル時の定数（constexpr）を用いることで、種組成を定義し、アプリケーションコードのコンパイル時間を大幅に短縮すること。

結果と応用
本ライブラリは、いくつかの主要な天体物理シミュレーションコードへの統合に成功している：

• Castro: ノヴァ、X線バースト、熱核爆発型超新星、および大質量星の対流のシミュレーションに使用されている。
• MAESTROeX: コンベクティブ・ウラカ過程およびX線バーストの研究に適用されている。
• Quokka: 特にライブラリのGPU対応反応ネットワーク積分器により採用されている。
• 機械学習: 本ライブラリは、核反応をモデル化するために深層ニューラルネットワークを訓練するという、近年の天体物理学における機械学習の研究を促進している。

意義
本論文は、本ライブラリをAMReXエコシステムにおける重要なインフラストラクチャ・コンポーネントとして位置づけており、冗長な開発努力を最小限に抑え、天体物理シミュレーションにおけるマイクロフィシックスへのアプローチを標準化することを目指している。堅牢でGPUに最適化された基盤を提供することにより、本ライブラリは、恒星対流から爆発的元素合成に至るまで、複雑で多次元的な問題に、より効率的に取り組むことを可能にする。著者らは、本ライブラリが同様の最近の取り組み（singularity-EOSなど）に先んじていること、およびAMReXライブラリ上に直接構築されたコードに特化した、より広範な物理範囲を提供していることを強調している。様々な機関の貢献者からなる開発チームは、オープンな科学的コラボレーションとして運営されており、分野の進化するニーズに対してライブラリが応答し続けられるようにしている。