Use QUDA for lattice QCD calculation with Python
Fuente:
arXiv
Enregistré dans:
| Auteurs principaux: | , , , , |
|---|---|
| Format: | Preprint |
| Publié: |
2024
|
| Sujets: | |
| Accès en ligne: | |
| Tags: |
Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
|
| _version_ | 1866910696405467136 |
|---|---|
| author | Jiang, Xiangyu Shi, Chunjiang Chen, Ying Gong, Ming Yang, Yi-Bo |
| author_facet | Jiang, Xiangyu Shi, Chunjiang Chen, Ying Gong, Ming Yang, Yi-Bo |
| contents | We developed PyQUDA, a Python wrapper for QUDA written in Cython, designed to facilitate lattice QCD calculations using the Python programming language. PyQUDA leverages the optimized linear algebra capabilities of NumPy/CuPy/PyTorch, along with the highly optimized lattice QCD operations provided by QUDA to accelerate research. This integration simplifies the process of writing calculation codes, enabling researchers to build more complex Python packages like EasyDistillation for specific physics objectives. PyQUDA supports a range of lattice QCD operations, including hybrid Monte Carlo (HMC) with N-flavor clover/HISQ fermions and inversion for the Wilson/clover/HISQ fermion action with the multigrid solver. It also includes utility functions for reading lattice QCD data stored in Chroma, MILC, and $χ$QCD formats. Type hints are supported by stub files and multi-GPU support is provided through mpi4py. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2411_08461 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Use QUDA for lattice QCD calculation with Python Jiang, Xiangyu Shi, Chunjiang Chen, Ying Gong, Ming Yang, Yi-Bo High Energy Physics - Lattice We developed PyQUDA, a Python wrapper for QUDA written in Cython, designed to facilitate lattice QCD calculations using the Python programming language. PyQUDA leverages the optimized linear algebra capabilities of NumPy/CuPy/PyTorch, along with the highly optimized lattice QCD operations provided by QUDA to accelerate research. This integration simplifies the process of writing calculation codes, enabling researchers to build more complex Python packages like EasyDistillation for specific physics objectives. PyQUDA supports a range of lattice QCD operations, including hybrid Monte Carlo (HMC) with N-flavor clover/HISQ fermions and inversion for the Wilson/clover/HISQ fermion action with the multigrid solver. It also includes utility functions for reading lattice QCD data stored in Chroma, MILC, and $χ$QCD formats. Type hints are supported by stub files and multi-GPU support is provided through mpi4py. |
| title | Use QUDA for lattice QCD calculation with Python |
| topic | High Energy Physics - Lattice |
| url | https://arxiv.org/abs/2411.08461 |