pyscf/gpu4pyscf

A plugin to use Nvidia GPU in PySCF package

해결하는 문제

PySCF(화학 시뮬레이션을 위한 파이썬 기반 프레임워크)에 GPU 가속 기능을 제공합니다. 이로 인해 하트리-폭크 및 밀도함수이론(DFT)과 같은 계산 비용이 큰 양자화학 계산이 NVIDIA GPU에 중량 수학 작업을 위임함으로써 크게 빨라집니다.

작동 방식

PySCF의 플러그인으로 작동하며, .to_gpu() 메서드를 사용해 PySCF 객체를 GPU로 이동할 수 있습니다. CUDA 호환 하드웨어와 CuPy, cuTENSOR 같은 라이브러리를 활용하여 전자 구조 계산에 필수적인 텐서 결합 및 기타 선형 대수 연산을 가속화합니다.

대상 사용자

에너지 계산, 해석적 기울기, 헤시안 행렬 계산에 소요되는 시간을 줄이고자 하는 전자 구조 시뮬레이션을 수행하는 계산 화학자 및 연구자들입니다.

주요 특징

  • 광범위한 기능 지원: libXC을 통해 LDA, GGA, mGGA, 하이브리드, 범위 분리 함수를 지원합니다.
  • 포괄적인 화학 도구: 해석적 기울기, 헤시안, 기하 최적화, 전이 상태 탐색을 지원합니다.
  • 고급 모델: PCM 및 SMD 용매 모델, 분산 보정(DFTD3/DFTD4)을 구현합니다.
  • 고성능: 다양한 분자 시스템에서 CPU 기반 대안 대비 뚜렷한 성능 향상을 보여줍니다.

관련

  • 프로젝트
  • 프로젝트
  • 프로젝트
  • 프로젝트
  • 프로젝트