Saved in:
| Main Authors: | , , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2401.01921 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866916655610724352 |
|---|---|
| author | Wu, Kai-Hsin Lin, Chang-Teng Hsu, Ke Hung, Hao-Ti Schneider, Manuel Chung, Chia-Min Kao, Ying-Jer Chen, Pochung |
| author_facet | Wu, Kai-Hsin Lin, Chang-Teng Hsu, Ke Hung, Hao-Ti Schneider, Manuel Chung, Chia-Min Kao, Ying-Jer Chen, Pochung |
| contents | We introduce a tensor network library designed for classical and quantum physics simulations called Cytnx (pronounced as sci-tens). This library provides almost an identical interface and syntax for both C++ and Python, allowing users to effortlessly switch between two languages. Aiming at a quick learning process for new users of tensor network algorithms, the interfaces resemble the popular Python scientific libraries like NumPy, Scipy, and PyTorch. Not only multiple global Abelian symmetries can be easily defined and implemented, Cytnx also provides a new tool called Network that allows users to store large tensor networks and perform tensor network contractions in an optimal order automatically. With the integration of cuQuantum, tensor calculations can also be executed efficiently on GPUs. We present benchmark results for tensor operations on both devices, CPU and GPU. We also discuss features and higher-level interfaces to be added in the future. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2401_01921 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | The Cytnx Library for Tensor Networks Wu, Kai-Hsin Lin, Chang-Teng Hsu, Ke Hung, Hao-Ti Schneider, Manuel Chung, Chia-Min Kao, Ying-Jer Chen, Pochung Mathematical Software Strongly Correlated Electrons We introduce a tensor network library designed for classical and quantum physics simulations called Cytnx (pronounced as sci-tens). This library provides almost an identical interface and syntax for both C++ and Python, allowing users to effortlessly switch between two languages. Aiming at a quick learning process for new users of tensor network algorithms, the interfaces resemble the popular Python scientific libraries like NumPy, Scipy, and PyTorch. Not only multiple global Abelian symmetries can be easily defined and implemented, Cytnx also provides a new tool called Network that allows users to store large tensor networks and perform tensor network contractions in an optimal order automatically. With the integration of cuQuantum, tensor calculations can also be executed efficiently on GPUs. We present benchmark results for tensor operations on both devices, CPU and GPU. We also discuss features and higher-level interfaces to be added in the future. |
| title | The Cytnx Library for Tensor Networks |
| topic | Mathematical Software Strongly Correlated Electrons |
| url | https://arxiv.org/abs/2401.01921 |