TorchSim/torch-sim

해결하는 문제

TorchSim은 원자론적 시뮬레이션을 가속화하기 위해 설계되었으며, 특히 머신러닝 원자간 포텐셜(MLIPs)을 사용할 때 유효합니다. 기존 도구들은 종종 MLIPs의 계산 오버헤드로 인해 어려움을 겪지만, TorchSim은 GPU 가속과 효율적인 메모리 관리를 활용하여 이러한 시뮬레이션을 훨씬 더 빠르게 실행할 수 있는 방법을 제공합니다.

작동 방식

원자론적 시뮬레이션의 핵심 프리미티브를 PyTorch로 직접 다시 작성함으로써, 이 엔진은 여러 시스템의 자동 배치(automatic batching)와 네이티브 GPU 지원을 가능하게 합니다. 이를 통해 많은 원자 시스템을 동시에 시뮬레이션할 수 있으며, 이는 ASE와 같은 기존 프레임워크와 비교하여 최대 100배의 속도 향상을 가져올 수 있습니다.

대상 사용자

이 도구는 분자 역학 또는 구조 완화(structure relaxation)를 수행하기 위해 MLIP 모델(MACE, Fairchem, SevenNet, MatterSim 등)을 사용하는 재료 과학 및 화학 분야의 연구자와 과학자들을 대상으로 합니다.

주요 특징

• 고속 MLIP 지원: 다양한 인기 MLIP 모델과의 네이티브 통합.
• 자동 배치(Automatic Batching): GPU 활용도를 극대화하기 위해 여러 시뮬레이션을 동시에 실행하는 기능.
• nComprehensive Simulation Tools: NVE, NVT, NPT와 같은 MD 통합 스킴과 FIRE, gradient descent와 같은 완화 방법, 그리고 Swap Monte Carlo 알고리즘을 포함합니다.
• Ecosystem Integration: ASE, Pymatgen, 그리고 Phonopy와 같은 일반적인 도구와 연동됩니다.
• Differentiable Simulation: 미분 가능한 시뮬레이션 및 탄성 특성 계산을 지원합니다.