ORB5X v1.0: a performance-portable global electromagnetic gyrokinetic PIC code in C++/Kokkos built using Agentic AI

核融合の偉大なる探索と、言語の壁

地球上で星を作ろうとしているところを想像してみてください。それが核融合の夢であり、太陽のエネルギー源でもあるプロセスであり、クリーンで無限のエネルギーを約束するものです。これを実現するために、科学者たちは「トカマク」と呼ばれる磁気ケージの中に、超高温のプラズマ（電荷を帯びた粒子の渦巻くスープ）を閉じ込めます。しかし、このプラズマは混沌としています。それは、まるで数十億匹の小さな、非常に活動的なハチが、互いに衝突し続け、方向を変え続ける様子を予測しようとするようなものです。このプラズマを安定させ、核融合に必要な温度を維持する方法を理解するために、科学者たちは大規模なコンピュータ・シミュレーションを使用します。これらのシミュレーションは仮想的な研究所として機能し、粒子がどのように動き、電磁場がどのようにねじれ、回転するかを追跡します。

この仕事において最も信頼されているツールの1つが、「ORB5」と呼ばれるコードです。ORB5を、数十年にわたって同じ完璧な核融合のレシピを作り続けてきた、熟練した古風なマスターシェフだと考えてください。しかし、このシェフは一つの言語しか話せません。それは、1950年代に発明されたプログラミング言語である「Fortran」です。Fortranは数学的な計算には優れていますが、まるでヴィンテージカーのようになりつつあります。修理が難しく、アップグレードが困難で、現代の道路（最新の高速なグラフィックスカードを搭載した最新のスーパーコンピュータなど）を走らせるのが難しいのです。コンピューティングの世界が新しいハードウェアへと移行する中で、この「ヴィンテージカー」を効率的に動かし続けることは困難になりつつあります。問題は、そのレシピが良いかどうかではありません。レシピは素晴らしいのです。問題は、料理の味を変えることなく、いかにしてそのレシピを現代の言語へと翻訳し、今日の最も速いエンジンで走らせることができるか、ということなのです。

翻訳プロジェクト：ORB5X

本論文では、その有名な核融合コードの野心的な新バージョンである「ORB5X」を紹介します。チームは新しいレシピを考案したり、プラズマの物理学を変更したりしようとしたわけではありません。代わりに、彼らは熟練した翻訳者として振る舞いました。彼らは元のFortranコードを取り上げ、「エージェンティックAI（コードを記述しデバッグするスマートなコンピュータ・アシスタント）」の助けを借りて、プログラム全体を現代的なプログラミング言語である「C++」へと書き換えました。

なぜこれが重要なのでしょうか？ 例えば、素晴らしい家の設計図を持っているのに、それが色褪せていて、特定の種類の製図台にしか適合しない紙に描かれている状況を想像してください。ORB5Xは、その全く同じ設計図を、あらゆるコンピュータやタブレット、あるいは宇宙にあるスーパーコンピュータでも動作するデジタルタブレット上に描き直すようなものです。目標は、元のコードの「数学的な魂」を保持しながら、それを「パフォーマンス・ポータブル（性能の移植性）」にすることでした。これは、新しいコードが標準的なノートパソコン、強力なデスクトップ、あるいは数千のグラフィックスカードを備えた巨大なスーパーコンピュータのいずれにおいても、それぞれのマシン用に完全に書き直すことなく、同様に優れたパフォーマンスで動作できることを意味します。

彼らがどのように行ったか：AIコパイロット

この翻訳は、単なるコピー＆ペースト作業ではありませんでした。それは慎重な、ステップ・バイ・ステップの再構築でした。研究者たちは、人間の専門家とAIエージェントが協力し合うワークフローを用いました。AIは疲れを知らないコパイロット（副操縦士）として機能しました。AIは古いFortranコードの塊を読み取り、それを現代的なC++でどのように記述すべきかを提案し、数学が狂う可能性のある潜在的な罠をフラグ立てしました。そして、人間の科学者がすべての提案をレビューし、新しいコードが元のコードと全く同じように振る舞うことを確認しました。

彼らは、二言語併用辞典のように、2つのバージョンを並べて管理しました。もし古いコードに fields.F90 というファイルがあれば、新しいコードには fields.cpp がありました。これにより、あらゆる計算をチェックすることができました。彼らは「Kokkos」と呼ばれる現代的なツールキットを使用しました。これはユニバーサル・アダプターのような役割を果たします。これによって、プログラマーが各デバイスに対して個別の指示を書くことなく、コードが異なる種類のコンピュータ・プロセッサ（Intel CPUやNVIDIA GPUなど）と自動的に通信できるようになります。

結果：マシン・アキュラシー（機械精度）

チームは、プラズマの乱流の異なる形態をシミュレートする3つの有名な「ストレス・テスト」を用いて、ORB5Xの実力を試しました。

1. ITG (イオン温度勾配): プラズマを通じて熱がどのように移動するかをシミュレートします。
2. ITPA: 高速で移動する粒子と磁気波が関与する複雑なテストです。
3. チャープ (Chirping): エネルギー粒子と相互作用する際の波の周波数がどのように変化するかを観察します。

結果は驚くべきものでした。新しいC++コードの出力をオリジナルのFortranコードと比較したところ、数値は「マシン・アキュラシー（機械精度）」で一致しました。これは、両者の差が極めて微小であり、コンピュータの計算精度による限界を除けば、実質的にゼロであることを意味します。シミュレーションを個人のノートパソコンで実行しても、LUMIやCINECAのような巨大なスーパーコンピュータで実行しても、物理学は同一のままでした。

なぜこれが大きなニュースなのか

この論文は、ORB5Xが成功であることを裏付けていますが、同時に現実的な期待も設定しています。新しいコードは、必ずしもすべてのマシンにおいて元のFortran版よりも速いわけではありません。実際、オリジナルのコードは数十年にわたって微調整されてきたため、一部のケースでは依然としてわずかに高速です。しかし、真の勝利は「移植性」にあります。オリジナルのFortranコードは、新しいハードウェアで実行するのが困難な「レガシー」システムになりつつあります。現代的なツールで構築されたORB5Xは、個人のノートパソコン（Apple M4チップを搭載したMacを含む）や、世界で最も高度なスーパーコンピュータ上で簡単にコンパイルして実行できます。

著者らは、これがバージョン1.0であることを注意深く述べています。これは強固な基礎であり、完成品ではありません。彼らは将来、コンピュータのメモリとプロセッサ間のデータ移動時間を短縮することで、さらに高速化することを計画しており、将とう的には、より複雑な形状の磁気ケージ（ステラレータなど）のシミュレーションに使用することを目指しています。しかし現時点では、数十年前の非常に複雑な科学コードを、AIの助けを借りて現代的な言語へと翻訳し、物理学を完璧に維持したまま実行できることを、彼らは見事に証明したのです。これは、過去と核融合研究の未来を繋ぐ架け橋であり、クリーンエネルギーへの探求が、世界が提供できる最も速いエンジン上で走り続けられることを保証するものなのです。