NVIDIA Vera CPUがロスアラモス国立研究所でのエージェンティック科学AIの道を開く

ロスアラモス国立研究所（LANL）は、HPEおよびNVIDIAと協力して、Mission、Vision、Veritasという3台の新しいスーパーコンピュータを構築しています。これらのシステムはNVIDIA Vera CPUを搭載し、科学的発見を加速し、科学分野におけるエージェンティックAIの可能性を切り開きます。

スーパーコンピュータは、HPE Cray Supercomputing GX5000アーキテクチャとNVIDIA Vera Rubinプラットフォームを採用し、NVIDIA Vera CPU、NVIDIA Rubin GPU、NVIDIA Quantum-X800 InfiniBandネットワーキングを組み合わせます。計画では、MissionにはNVIDIA Vera Rubin GPUノードと2,300台のスタンドアロンNVIDIA Vera CPU（HPE Cray GX240ブレード使用）が含まれます。Veritasは約1,150台のスタンドアロンNVIDIA Vera CPUを備え、Vera Rubinノードを補完し、MissionやVisionとともに研究所主導の研究開発プログラムを支援し、科学におけるエージェンティックAIを加速します。

研究者たちは、仮説の立案、ツールの選択、シミュレーションの実行、出力の分析、次のステップの洗練を行うことができるAIエージェントを科学の新たなツールとして追加しています。LANLのURSA（汎用研究科学エージェント）に関する公開作業は、Venadoで実行され、まもなくMissionやVisionでも実行される予定で、この方向性を示しています。これは、科学者が仮説を考え、実験を計画し、シミュレーションを実行し、結果を分析するのを支援するために設計された、モジュール型でフィードバック駆動のAIフレームワークです。

LANLのテストでは、Vera CPUはURSAワークロードにおいてCrossroads x86スーパーコンピュータのCPUよりも7倍高い性能を発揮しました。Branson（オープンソースのモンテカルロ熱伝達シミュレーションツール）の初期テストでは、Vera CPUはCrossroads x86 CPUの3倍以上の性能を示しました。これらの結果は、VeraのカスタムOlympusコア、LPDDR5メモリ、高速オンチップファブリックによって可能になりました。単一のVera CPUは、シングルソケットのx86 CPUよりも3倍以上高い性能を提供し、コアあたりのメモリ帯域幅は4倍以上、ノードあたりのメモリ容量は6倍以上であり、LANLでのより迅速な科学的結果を実現します。

研究所のすべてのスーパーコンピュータは、ハードウェアアーキテクト、システムソフトウェア開発者、ドメイン科学者、コンピュータ科学者、応用数学者によって共同設計されており、抽象的なベンチマークだけでなく、実際の科学的ワークロードによってシステムが形成されることを保証しています。

Missionは2027年に運用開始予定で、国家核安全保障局の高度シミュレーション・コンピューティングプログラムにおける5番目の先端技術システムとなり、機密の国家安全保障ワークロードにおいてCrossroadsを置き換えます。Visionも2027年に運用開始予定で、材料科学、核科学、エネルギー modeling、生物医学研究、AIなどの基礎科学研究のリソースとして機能し、より多くの科学者がリスクの高い作業に移行する前に手法をテストし、モデルを訓練し、アイデアを探求できるようにします。

この取り組みは、GraceからVeraに至るまで、LANLとNVIDIAがCPUに関して10年以上にわたって深く協力してきた成果を拡張するものであり、LANLのシミュレーションワークロードに合わせた極端なコデザインを採用しています。3台の新しいスーパーコンピュータは、2024年にロスアラモスに設置されたHPE Cray EXスーパーコンピュータVenadoを基盤としており、NVIDIA GH200 Grace HopperスーパーチップとNVIDIA Grace CPUスーパーチップを搭載しています。

NVIDIA Vera CPUの詳細についてはこちらをご覧ください。