[논문 리뷰] GPAW: An open Python package for electronic-structure calculations

Jens Jørgen Mortensen, Ask Hjorth Larsen외 45명|arXiv (Cornell University)|2023. 10. 23.

Physics of Superconductivity and Magnetism인용 수 11

한 줄 요약

GPAW은 PAW를 사용한 전자구조 계산을 위한 Python 기반의 다중 표현 오픈소스 패키지입니다; 실공간 격자, 평면파, 수치 원자 궤도함수를 지원하며, DFt를 넘는 방법들과 GPU 가속을 제공합니다.

ABSTRACT

We review the GPAW open-source Python package for electronic structure calculations. GPAW is based on the projector-augmented wave method and can solve the self-consistent density functional theory (DFT) equations using three different wave-function representations, namely real-space grids, plane waves, and numerical atomic orbitals. The three representations are complementary and mutually independent and can be connected by transformations via the real-space grid. This multi-basis feature renders GPAW highly versatile and unique among similar codes. By virtue of its modular structure, the GPAW code constitutes an ideal platform for implementation of new features and methodologies. Moreover, it is well integrated with the Atomic Simulation Environment (ASE) providing a flexible and dynamic user interface. In addition to ground-state DFT calculations, GPAW supports many-body GW band structures, optical excitations from the Bethe-Salpeter Equation (BSE), variational calculations of excited states in molecules and solids via direct optimization, and real-time propagation of the Kohn-Sham equations within time-dependent DFT. A range of more advanced methods to describe magnetic excitations and non-collinear magnetism in solids are also now available. In addition, GPAW can calculate non-linear optical tensors of solids, charged crystal point defects, and much more. Recently, support of GPU acceleration has been achieved with minor modifications of the GPAW code thanks to the CuPy library. We end the review with an outlook describing some future plans for GPAW.

연구 동기 및 목표

• GPAW 소프트웨어와 그 설계 철학에 대한 개요를 제공합니다.
• 세 가지 파동함수 표현(real-space grid, plane waves, LCAO)와 그 보완성에 대해 제시합니다.
• 바닥상태 DFT 기능과 PAW 구현을 설명합니다.
• beyond-DFT 방법 및 excited-state 기능(GW, BSE, TDDFT)을 강조합니다.
• 개발자 및 사용자 관점, 성능, 그리고 미래 방향성을 설명합니다.

제안 방법

• PAW 형식주의와 GPAW가 T 연산자를 통해 모든 전자(all-electron)를 의사 공간(pseudo-space)으로 매핑하는 방법을 설명합니다.
• 세 가지 파동함수 표현(PW, FD real-space grid, LCAO)과 이들 간의 변환이 어떻게 연결되는지 설명합니다.
• 코른–샴(Kohn–Sham) 방정식 해를 구하는 방법들(diagonalization with density mixing, direct minimization)을 개요합니다.
• 병렬화 전략, GPU 구현, 그리고 라이브러리(F FFTW, ScaLAPACK, ELPA, Libxc, libvdwxc) 와의 인터페이스를 논의합니다.
• GPAW가 고급 방법들(GW, BSE, TDDFT, 비정렬 자기 현상) 및 기타 기능들(Berry 위상, Wannier 함수)을 어떻게 가능하게 하는지 설명합니다.
• 소프트웨어 개발 결정, 테스트, 새 코드베이스와 구 코드베이스의 전환을 설명합니다.

Figure 2: Bayesian ensemble of XC functionals around BEEF–vdW. Orange solid line is the BEEF–vdW exchange enhancement factor, while the blue lines depict $F_{\text{x}}(s)$ for 50 samples of the randomly generated ensemble. Dotted black lines mark the exchange model perturbations that yield DFT resul

실험 결과

연구 질문

• RQ1 GPAW가 전자구조 문제를 해결하기 위해 여러 기저 표현을 어떻게 통합합니까?
• RQ2 GPAW가 어떤 beyond-DFT 및 excited-state 기능을 지원하며 어떤 표현에서인가?
• RQ3 PAW가 어떻게 구현되며 GPAW 내에서 모든-electron 양이 어떻게 얻어지나요?
• RQ4 GPAW의 성능, 확장성 및 GPU 가속 측면은 어떤가요?
• RQ5 GPAW의 설계가 사용자 맞춤화 및 코드 개발을 어떻게 촉진할 수 있나요?

주요 결과

• GPAW은 평면파(PW), 실공간 격자(real-space grids), 그리고 LCAO의 세 가지 보완적 파동함수 표현을 실공간 격자를 통해 변환 가능하도록 지원합니다.
• PAW 방법은 의사 표현으로부터 all-electron 양을 접근할 수 있도록 구현되어 있습니다.
• Beyond-DFT 방법 such as GW and BSE, as well as TDDFT and spin–orbit coupling, are available within GPAW, particularly in PW mode.
• GPU acceleration has been achieved for PW mode using CuPy with minimal code changes.
• GPAW의 integration with ASE provides a flexible interface and supports direct access to core quantities for analysis and scripting.
• A new ground-state DFT code design is being developed to improve feature implementation and maintainability, with new features already deployed (GPU PW, spin spirals).

Figure 3: Band structure of WS 2 monolayer obtained from PBE with non-selfconsistent spin–orbit coupling. The colors indicate the expectation value of $S_{z}$ for each state. The grey lines show the band structure without spin–orbit coupling.