TorchSim/torch-sim

What it solves

TorchSimは、特に機械学習原子間ポテンシャル（MLIPs）を使用する場合において、原子論的シミュレーションを加速させるために設計されています。従来のツールはMLIPsの計算オーバーヘッドに苦労することが多いですが、TorchSimはGPU加速と効率的なメモリ管理を活用することで、これらのシミュレーションを大幅に高速化する方法を提供します。

How it works

原子論的シミュレーションのコアプリミティブをPyTorchで直接書き直すことにより、このエンジンは複数のシステムの自動バッチ処理とネイティブなGPUサポートを可能にします。これにより、多くの原子系を同時にシミュレーションすることができ、ASEのような従来のフレームワークと比較して最大100倍の高速化を実現できます。

Who it’s for

これは、分子動力学や構造緩和を行うためにMLIPモデル（MACE、Fairchem、SevenNet、およびMatterSim）を使用する、材料科学や化学分野の研究者や科学者を対象としています。

Highlights

• High-speed MLIP support: 幅広い人気のあるMLIPモデルとのネイティブな統合。
• Automatic Batching: GPU利用率を最大化するために、複数のシミュレーションを同時に実行する能力。
• Comprehensive Simulation Tools: MD統合スキーム（NVE、NVT、NPT）、緩和方法（FIRE、gradient descent）、およびSwap Monte Carloアルゴリズムを含みます。
• Ecosystem Integration: ASE、Pymatgen、およびPhonopyのような一般的なツールと連携します。
• Differentiable Simulation: 微分可能なシミュレーションと弾性特性の計算をサポートします。