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와 같은 맞춤형 확장 패키지를 허용하는 확장 가능한 아키텍처.