PYATB: An Efficient Python Package for Electronic Structure Calculations Using Ab Initio Tight-Binding Model

Gan Jin,ongsheng Pang,Yuyang Ji,Zujian Dai,Lixin He
DOI: https://doi.org/10.1016/j.cpc.2023.108844
2023-03-31
Abstract:We present PYATB, a Python package designed for computing band structures and related properties of materials using the ab initio tight-binding Hamiltonian. The Hamiltonian is directly obtained after conducting self-consistent calculations with first-principles packages using numerical atomic orbital (NAO) bases, such as ABACUS. The package comprises three modules: Bands, Geometric, and Optical. In the Bands module, one can calculate essential properties of band structures, including the partial density of states (PDOS), fat bands, Fermi surfaces, and Weyl/Dirac points. The band unfolding method is utilized to obtain the energy band spectra of a supercell by projecting the electronic structure of the supercell onto the Brillouin zone of the primitive cell. With the Geometric module, one can compute the Berry phase and Berry curvature-related quantities, such as electric polarization, Wilson loops, Chern numbers, and anomalous Hall conductivities. The Optical module offers a range of optical property calculations, including optical conductivity and nonlinear optical responses, such as shift current and Berry curvature dipole.
Materials Science
What problem does this paper attempt to address?
### What problems does this paper attempt to solve? This paper introduced a Python software package named PYATB, which aims to efficiently calculate the electronic structure of materials and their related properties through the ab initio tight - binding model. Specifically, PYATB attempts to solve the following key problems: 1. **Efficient calculation of electronic structure**: - The paper points out that although traditional first - principles calculations based on plane waves are accurate, they require a large number of k - points, resulting in very high computational costs. PYATB uses numerical atomic orbitals (NAO) as a basis and directly generates ab initio tight - binding Hamiltonians from self - consistent calculations, avoiding the process of constructing maximally - localized Wannier functions (MLWF), thereby improving computational efficiency. 2. **Calculation of band geometry and topological properties**: - PYATB can calculate physical quantities related to the Berry phase and Berry curvature, such as electric polarization, Wilson loops, Chern numbers, and anomalous Hall conductivity. These quantities are crucial for understanding materials with non - trivial topological properties, such as topological insulators, Dirac/Weyl semimetals. 3. **Calculation of optical properties**: - PYATB can also calculate linear and nonlinear optical responses, including optical conductivity, dielectric functions, shift currents, and Berry curvature dipoles. This is very useful for studying the photoelectric effects of materials under light irradiation. 4. **Handling supercell structures**: - For imperfect crystals containing defect, doping, or alloy atoms, supercell approximation is usually used for first - principles calculations. However, the band structure of the supercell will be severely folded in the Brillouin zone, making it difficult to analyze. PYATB implements the band unfolding method, which can project the band structure of the supercell onto the Brillouin zone of the primitive cell, facilitating comparison with angle - resolved photoemission spectroscopy (ARPES) experimental results. ### Summary The main goal of PYATB is to provide an efficient and feature - rich tool for calculating the electronic structure, band geometry and topological properties, and optical responses of solid materials. By combining the ab initio tight - binding model and numerical atomic orbital basis, PYATB can significantly improve computational efficiency while maintaining computational accuracy, and is especially suitable for studying materials with complex topological properties and optical characteristics.