mir-group/nequip

NequIP is a code for building E(3)-equivariant interatomic potentials

What it solves

NequIP は、系内の原子のエネルギーと力をモデル化するために使用される E(3) エクイバリアントな原子間ポテンシャルを構築するフレームワークを提供します。これらの深層エクイバリアントモデルに対して高性能なトレーニングと推論を実現し、分子シミュレーションにおけるデータ効率と精度を向上させることを目的としています。

How it works

NequIP は E(3) エクイバリアントなグラフニューラルネットワークを用いて、原子系の幾何学的特性を表現します。柔軟なフレームワークとして設計されており、ユーザーはカスタムアーキテクチャを構築したり新しいトレーニング手法を実装したりできます。また、ASE（Atomic Simulation Environment）や LAMMPS などの一般的な科学ソフトウェアと統合され、分子動力学シミュレーションに利用できます。

Who it’s for

計算化学、材料科学、バイオ分子シミュレーションの研究者で、原子スケールの相互作用をシミュレートするために正確でデータ効率の高い原子間ポテンシャルが必要な方々向けです。

Highlights

  • コンパイルされたトレーニングと推論により速度向上。
  • マルチ GPU トレーニングをサポート。
  • OpenEquivariance と CuEquivariance による GPU カーネルアクセラレーション。
  • ASE カルキュレータおよび LAMMPS ペアスタイルとの統合。
  • Allegro などのカスタム拡張パッケージを可能にする拡張性のあるアーキテクチャ。