tensorcircuit

Tensor network based quantum software framework for the NISQ era

https://github.com/tencent-quantum-lab/tensorcircuit

Science Score: 77.0%

This score indicates how likely this project is to be science-related based on various indicators:

  • CITATION.cff file
    Found CITATION.cff file
  • codemeta.json file
    Found codemeta.json file
  • .zenodo.json file
    Found .zenodo.json file
  • DOI references
    Found 2 DOI reference(s) in README
  • Academic publication links
    Links to: arxiv.org, nature.com
  • Committers with academic emails
    5 of 30 committers (16.7%) from academic institutions
  • Institutional organization owner
  • JOSS paper metadata
  • Scientific vocabulary similarity
    Low similarity (16.1%) to scientific vocabulary

Keywords

automatic-differentiation jax machine-learning matrix-product-states neural-network nisq open-quantum-systems pytorch quantum quantum-algorithms quantum-circuit quantum-computing quantum-dynamics quantum-error-mitigation quantum-machine-learning quantum-noise quantum-simulation tensor-network tensorflow variational-quantum-algorithms

Keywords from Contributors

degoogle
Last synced: 6 months ago · JSON representation ·

Repository

Tensor network based quantum software framework for the NISQ era

Basic Info
Statistics
  • Stars: 326
  • Watchers: 7
  • Forks: 90
  • Open Issues: 23
  • Releases: 20
Topics
automatic-differentiation jax machine-learning matrix-product-states neural-network nisq open-quantum-systems pytorch quantum quantum-algorithms quantum-circuit quantum-computing quantum-dynamics quantum-error-mitigation quantum-machine-learning quantum-noise quantum-simulation tensor-network tensorflow variational-quantum-algorithms
Created almost 4 years ago · Last pushed 6 months ago
Metadata Files
Readme Changelog Contributing License Code of conduct Citation

README.md

English | 简体中文

TensorCircuit is the next generation of quantum software framework with support for automatic differentiation, just-in-time compiling, hardware acceleration, and vectorized parallelism.

TensorCircuit is built on top of modern machine learning frameworks: Jax, TensorFlow, and PyTorch. It is specifically suitable for highly efficient simulations of quantum-classical hybrid paradigm and variational quantum algorithms in ideal, noisy and approximate cases. It also supports real quantum hardware access and provides CPU/GPU/QPU hybrid deployment solutions since v0.9.

Getting Started

Please begin with Quick Start in the full documentation.

For more information on software usage, sota algorithm implementation and engineer paradigm demonstration, please refer to 70+ example scripts and 30+ tutorial notebooks. API docstrings and test cases in tests are also informative.

The following are some minimal demos.

  • Circuit manipulation:

python import tensorcircuit as tc c = tc.Circuit(2) c.H(0) c.CNOT(0,1) c.rx(1, theta=0.2) print(c.wavefunction()) print(c.expectation_ps(z=[0, 1])) print(c.sample(allow_state=True, batch=1024, format="count_dict_bin"))

  • Runtime behavior customization:

python tc.set_backend("tensorflow") tc.set_dtype("complex128") tc.set_contractor("greedy")

  • Automatic differentiations with jit:

```python def forward(theta): c = tc.Circuit(2) c.R(0, theta=theta, alpha=0.5, phi=0.8) return tc.backend.real(c.expectation((tc.gates.z(), [0])))

g = tc.backend.grad(forward) g = tc.backend.jit(g) theta = tc.arraytotensor(1.0) print(g(theta)) ```

More highlight features for TensorCircuit (click for details) - Sparse Hamiltonian generation and expectation evaluation: ```python n = 6 pauli_structures = [] weights = [] for i in range(n): pauli_structures.append(tc.quantum.xyz2ps({"z": [i, (i + 1) % n]}, n=n)) weights.append(1.0) for i in range(n): pauli_structures.append(tc.quantum.xyz2ps({"x": [i]}, n=n)) weights.append(-1.0) h = tc.quantum.PauliStringSum2COO(pauli_structures, weights) print(h) # BCOO(complex64[64, 64], nse=448) c = tc.Circuit(n) c.h(range(n)) energy = tc.templates.measurements.operator_expectation(c, h) # -6 ``` - Large-scale simulation with tensor network engine ```python # tc.set_contractor("cotengra-30-10") n=500 c = tc.Circuit(n) c.h(0) c.cx(range(n-1), range(1, n)) c.expectation_ps(z=[0, n-1], reuse=False) ``` - Density matrix simulator and quantum info quantities ```python c = tc.DMCircuit(2) c.h(0) c.cx(0, 1) c.depolarizing(1, px=0.1, py=0.1, pz=0.1) dm = c.state() print(tc.quantum.entropy(dm)) print(tc.quantum.entanglement_entropy(dm, [0])) print(tc.quantum.entanglement_negativity(dm, [0])) print(tc.quantum.log_negativity(dm, [0])) ```

Install

The package is written in pure Python and can be obtained via pip as:

python pip install tensorcircuit

We recommend you install this package with tensorflow also installed as:

python pip install tensorcircuit[tensorflow]

Other optional dependencies include [torch], [jax], [qiskit] and [cloud].

We also have Docker support.

Advantages

  • Tensor network simulation engine based

  • JIT, AD, vectorized parallelism compatible

  • GPU support, quantum device access support, hybrid deployment support

  • Efficiency

    • Time: 10 to 10^6+ times acceleration compared to TensorFlow Quantum, Pennylane or Qiskit
    • Space: 600+ qubits 1D VQE workflow (converged energy inaccuracy: < 1%)
  • Elegance

    • Flexibility: customized contraction, multiple ML backend/interface choices, multiple dtype precisions, multiple QPU providers
    • API design: quantum for humans, less code, more power
  • Batteries included

Tons of amazing features and built in tools for research (click for details)

  • Support super large circuit simulation using tensor network engine.

  • Support noisy simulation with both Monte Carlo and density matrix (tensor network powered) modes.

  • Support approximate simulation with MPS-TEBD modes.

  • Support analog/digital hybrid simulation (time dependent Hamiltonian evolution, pulse level simulation) with neural ode modes.

  • Support Fermion Gaussian state simulation with expectation, entanglement, measurement, ground state, real and imaginary time evolution.

  • Support qudits simulation.

  • Support parallel quantum circuit evaluation across multiple GPUs.

  • Highly customizable noise model with gate error and scalable readout error.

  • Support for non-unitary gate and post-selection simulation.

  • Support real quantum devices access from different providers.

  • Scalable readout error mitigation native to both bitstring and expectation level with automatic qubit mapping consideration.

  • Advanced quantum error mitigation methods and pipelines such as ZNE, DD, RC, etc.

  • Support MPS/MPO as representations for input states, quantum gates and observables to be measured.

  • Support vectorized parallelism on circuit inputs, circuit parameters, circuit structures, circuit measurements and these vectorization can be nested.

  • Gradients can be obtained with both automatic differenation and parameter shift (vmap accelerated) modes.

  • Machine learning interface/layer/model abstraction in both TensorFlow and PyTorch for both numerical simulation and real QPU experiments.

  • Circuit sampling supports both final state sampling and perfect sampling from tensor networks.

  • Light cone reduction support for local expectation calculation.

  • Highly customizable tensor network contraction path finder with opteinsum interface.

  • Observables are supported in measurement, sparse matrix, dense matrix and MPO format.

  • Super fast weighted sum Pauli string Hamiltonian matrix generation.

  • Reusable common circuit/measurement/problem templates and patterns.

  • Jittable classical shadow infrastructures.

  • SOTA quantum algorithm and model implementations.

  • Support hybrid workflows and pipelines with CPU/GPU/QPU hardware from local/cloud/hpc resources using tf/torch/jax/cupy/numpy frameworks all at the same time.

Contributing

Status

This project is created and maintained by Shi-Xin Zhang with current core authors Shi-Xin Zhang and Yu-Qin Chen. We also thank contributions from the open source community.

Citation

If this project helps in your research, please cite our software whitepaper to acknowledge the work put into the development of TensorCircuit.

TensorCircuit: a Quantum Software Framework for the NISQ Era (published in Quantum)

which is also a good introduction to the software.

Research works citing TensorCircuit can be highlighted in Research and Applications section.

Guidelines

For contribution guidelines and notes, see CONTRIBUTING.

We welcome issues, PRs, and discussions from everyone, and these are all hosted on GitHub.

License

TensorCircuit is open source, released under the Apache License, Version 2.0.

Contributors

Shixin Zhang
Shixin Zhang

💻 📖 💡 🤔 🚇 🚧 🔬 👀 🌍 ⚠️ 📢 💬
Yuqin Chen
Yuqin Chen

💻 📖 💡 🤔 🔬 ⚠️ 📢
Jiezhong Qiu
Jiezhong Qiu

💻 💡 🤔 🔬
Weitang Li
Weitang Li

💻 📖 🤔 🔬 ⚠️ 📢
Jiace Sun
Jiace Sun

💻 📖 💡 🤔 🔬 ⚠️
Zhouquan Wan
Zhouquan Wan

💻 📖 💡 🤔 🔬 ⚠️
Shuo Liu
Shuo Liu

💡 🔬
Hao Yu
Hao Yu

💻 📖 🚇 ⚠️
Xinghan Yang
Xinghan Yang

📖 🌍
JachyMeow
JachyMeow

🌍
Zhaofeng Ye
Zhaofeng Ye

🎨
erertertet
erertertet

💻 📖 ⚠️
Yicong Zheng
Yicong Zheng

Zixuan Song
Zixuan Song

📖 🌍 💻 ⚠️
Hao Xie
Hao Xie

📖
Pramit Singh
Pramit Singh

⚠️
Jonathan Allcock
Jonathan Allcock

📖 🤔 📢
nealchen2003
nealchen2003

📖
隐公观鱼
隐公观鱼

💻 ⚠️
WiuYuan
WiuYuan

💡
Felix Xu
Felix Xu

💻 ⚠️
Hong-Ye Hu
Hong-Ye Hu

📖
peilin
peilin

💻 ⚠️ 📖
Cristian Emiliano Godinez Ramirez
Cristian Emiliano Godinez Ramirez

💻 ⚠️
ztzhu
ztzhu

💻
Rabqubit
Rabqubit

💡
Kazuki Tsuoka
Kazuki Tsuoka

💻 ⚠️ 📖 💡
Gopal Ramesh Dahale
Gopal Ramesh Dahale

💡
Chanandellar Bong
Chanandellar Bong

💡

Research and Applications

DQAS

For the application of Differentiable Quantum Architecture Search, see applications.

Reference paper: https://arxiv.org/abs/2010.08561 (published in QST).

VQNHE

For the application of Variational Quantum-Neural Hybrid Eigensolver, see applications.

Reference paper: https://arxiv.org/abs/2106.05105 (published in PRL) and https://arxiv.org/abs/2112.10380 (published in AQT).

VQEX-MBL

For the application of VQEX on MBL phase identification, see the tutorial.

Reference paper: https://arxiv.org/abs/2111.13719 (published in PRB).

Stark-DTC

For the numerical demosntration of discrete time crystal enabled by Stark many-body localization, see the Floquet simulation demo.

Reference paper: https://arxiv.org/abs/2208.02866 (published in PRL).

RA-Training

For the numerical simulation of variational quantum algorithm training using random gate activation strategy by us, see the project repo.

Reference paper: https://arxiv.org/abs/2303.08154 (published in PRR as a Letter).

TenCirChem

TenCirChem is an efficient and versatile quantum computation package for molecular properties. TenCirChem is based on TensorCircuit and is optimized for chemistry applications.

Reference paper: https://arxiv.org/abs/2303.10825 (published in JCTC).

EMQAOA-DARBO

For the numerical simulation and hardware experiments with error mitigation on QAOA, see the project repo.

Reference paper: https://arxiv.org/abs/2303.14877 (published in Communications Physics).

NN-VQA

For the setup and simulation code of neural network encoded variational quantum eigensolver, see the demo.

Reference paper: https://arxiv.org/abs/2308.01068 (published in PRApplied).

More works

More research works and code projects using TensorCircuit (click for details)

  • Neural Predictor based Quantum Architecture Search: https://arxiv.org/abs/2103.06524 (published in Machine Learning: Science and Technology).

  • Quantum imaginary-time control for accelerating the ground-state preparation: https://arxiv.org/abs/2112.11782 (published in PRR).

  • Efficient Quantum Simulation of Electron-Phonon Systems by Variational Basis State Encoder: https://arxiv.org/abs/2301.01442 (published in PRR).

  • Variational Quantum Simulations of Finite-Temperature Dynamical Properties via Thermofield Dynamics: https://arxiv.org/abs/2206.05571.

  • Understanding quantum machine learning also requires rethinking generalization: https://arxiv.org/abs/2306.13461 (published in Nature Communications).

  • Decentralized Quantum Federated Learning for Metaverse: Analysis, Design and Implementation: https://arxiv.org/abs/2306.11297. Code: https://github.com/s222416822/BQFL.

  • Non-IID quantum federated learning with one-shot communication complexity: https://arxiv.org/abs/2209.00768 (published in Quantum Machine Intelligence). Code: https://github.com/JasonZHM/quantum-fed-infer.

  • Quantum generative adversarial imitation learning: https://doi.org/10.1088/1367-2630/acc605 (published in New Journal of Physics).

  • GSQAS: Graph Self-supervised Quantum Architecture Search: https://arxiv.org/abs/2303.12381 (published in Physica A: Statistical Mechanics and its Applications).

  • Practical advantage of quantum machine learning in ghost imaging: https://www.nature.com/articles/s42005-023-01290-1 (published in Communications Physics).

  • Zero and Finite Temperature Quantum Simulations Powered by Quantum Magic: https://arxiv.org/abs/2308.11616.

  • Comparison of Quantum Simulators for Variational Quantum Search: A Benchmark Study: https://arxiv.org/abs/2309.05924.

  • Statistical analysis of quantum state learning process in quantum neural networks: https://arxiv.org/abs/2309.14980 (published in NeurIPS).

  • Generative quantum machine learning via denoising diffusion probabilistic models: https://arxiv.org/abs/2310.05866 (published in PRL).

  • Quantum imaginary time evolution and quantum annealing meet topological sector optimization: https://arxiv.org/abs/2310.04291.

  • Google Summer of Code 2023 Projects (QML4HEP): https://github.com/ML4SCI/QMLHEP, https://github.com/Gopal-Dahale/qgnn-hep, https://github.com/salcc/QuantumTransformers.

  • Absence of barren plateaus in finite local-depth circuits with long-range entanglement: https://arxiv.org/abs/2311.01393 (published in PRL).

  • Non-Markovianity benefits quantum dynamics simulation: https://arxiv.org/abs/2311.17622.

If you want to highlight your research work or projects here, feel free to add by opening PR.

Owner

  • Name: Tencent Quantum Lab
  • Login: tencent-quantum-lab
  • Kind: organization
  • Email: qlab@tencent.com
  • Location: China

Tencent Quantum Lab

Citation (CITATION.cff)

cff-version: 1.2.0
message: "If you find this software helpful in your research, please cite it as below."
authors:
- family-names: "Zhang"
  given-names: "Shi-Xin"
- family-names: "Chen"
  given-names: "Yu-Qin"
title: "TensorCircuit"
version: 0.7.0
date-released: 2020-04-19
url: "https://github.com/tencent-quantum-lab/tensorcircuit"
preferred-citation:
  type: article
  authors:
  - family-names: "Zhang"
    given-names: "Shi-Xin"
  - family-names: "Allcock"
    given-names: "Jonathan"
  - family-names: "Wan"
    given-names: "Zhou-Quan"
  - family-names: "Liu"
    given-names: "Shuo"
  - family-names: "Sun"
    given-names: "Jiace"
  - family-names: "Yu"
    given-names: "Hao"
  - family-names: "Yang"
    given-names: "Xing-Han"
  - family-names: "Qiu"
    given-names: "Jiezhong"
  - family-names: "Ye"
    given-names: "Zhaofeng"
  - family-names: "Chen"
    given-names: "Yu-Qin"
  - family-names: "Lee"
    given-names: "Chee-Kong"
  - family-names: "Zheng"
    given-names: "Yi-Cong"
  - family-names: "Jian"
    given-names: "Shao-Kai"
  - family-names: "Yao"
    given-names: "Hong"
  - family-names: "Hsieh"
    given-names: "Chang-Yu"
  - family-names: "Zhang"
    given-names: "Shengyu"
  doi: "10.22331/q-2023-02-02-912"
  journal: "Quantum"
  month: 2
  pages: 912
  title: "TensorCircuit: a Quantum Software Framework for the NISQ Era"
  volume: 7
  year: 2023

GitHub Events

Total
  • Issues event: 2
  • Watch event: 48
  • Member event: 3
  • Issue comment event: 9
  • Push event: 6
  • Pull request event: 9
  • Fork event: 8
Last Year
  • Issues event: 2
  • Watch event: 48
  • Member event: 3
  • Issue comment event: 9
  • Push event: 6
  • Pull request event: 9
  • Fork event: 8

Committers

Last synced: 9 months ago

All Time
  • Total Commits: 1,615
  • Total Committers: 30
  • Avg Commits per committer: 53.833
  • Development Distribution Score (DDS): 0.381
Past Year
  • Commits: 28
  • Committers: 5
  • Avg Commits per committer: 5.6
  • Development Distribution Score (DDS): 0.536
Top Committers
Name Email Commits
refraction-ray k****l@f****m 999
Xinghan_Yang y****n@o****m 110
allcontributors[bot] 4****] 72
YHPeter y****3@h****m 56
yutuer21 5****9@q****m 53
PeilinZHENG p****g@g****m 49
JiaceSun j****3@c****u 47
Xinghan Yang 2****4@q****m 33
MarkSong535 7****5 31
Felix Xu 6****5 25
Kazuki Tsuoka k****a@g****p 19
xptree j****u@o****m 18
liwt31 l****1@1****m 13
Zhou-quan Wan 9****9@q****m 13
weitang li w****i@t****m 13
ls-iastu l****0@m****n 10
eurethia 3****1@q****m 9
WiuYuan 1****n 7
JachyMeow 1****w 7
pramitsingh0 p****0@g****m 5
Erertertet e****t@g****m 5
nealyxchen n****n@t****m 5
JAllcock j****k@g****m 3
Yuxuan y****n@d****m 2
EmilianoG-byte e****8@g****m 2
Gopal Dahale d****7@g****m 2
Hao Xie x****8@m****n 2
Hong-Ye Hu 5****u 2
ztzhu1 z****n@z****n 2
tencentqlab w****u@t****m 1

Packages

  • Total packages: 2
  • Total downloads:
    • pypi 503 last-month
  • Total dependent packages: 0
    (may contain duplicates)
  • Total dependent repositories: 2
    (may contain duplicates)
  • Total versions: 47
  • Total maintainers: 2
proxy.golang.org: github.com/tencent-quantum-lab/tensorcircuit
  • Versions: 20
  • Dependent Packages: 0
  • Dependent Repositories: 0
Rankings
Dependent packages count: 6.4%
Average: 6.7%
Dependent repos count: 6.9%
Last synced: 6 months ago
pypi.org: tensorcircuit

High performance unified quantum computing framework for the NISQ era

  • Versions: 27
  • Dependent Packages: 0
  • Dependent Repositories: 2
  • Downloads: 503 Last month
Rankings
Stargazers count: 4.6%
Forks count: 5.5%
Dependent packages count: 7.3%
Average: 8.3%
Dependent repos count: 11.8%
Downloads: 12.1%
Maintainers (2)
Last synced: 6 months ago

Dependencies

.github/nightly_build/setup.py pypi
  • numpy *
benchmarks/requirements.txt pypi
  • jax ==0.2.21
  • networkx *
  • pennylane ==0.18
  • py-cpuinfo *
  • scikit-learn *
  • sympy *
  • tensorflow ==2.4.1
  • tensorflow-quantum ==0.5.1
  • tensornetwork ==0.4.5
requirements/requirements-dev.txt pypi
  • black ==22.3.0 development
  • furo * development
  • mypy ==0.782 development
  • nbsphinx * development
  • numpy ==1.21.5 development
  • pylint ==2.11.1 development
  • pytest ==6.2.4 development
  • pytest-benchmark * development
  • pytest-cov * development
  • pytest-lazy-fixture * development
  • pytest-xdist * development
  • sphinx >=4.0 development
  • sphinx-copybutton * development
requirements/requirements-docker.txt pypi
  • baytune *
  • black ==22.3.0
  • cirq *
  • furo *
  • graphviz *
  • jax ==0.3.4
  • jupyter *
  • kahypar *
  • matplotlib *
  • mypy ==0.782
  • nbsphinx *
  • networkx *
  • nevergrad *
  • numpy *
  • openfermion *
  • openfermionpyscf *
  • optax *
  • optuna *
  • pennylane *
  • pylint *
  • pytest *
  • pytest-benchmark *
  • pytest-cov *
  • pytest-lazy-fixture *
  • pytest-xdist *
  • qiskit *
  • quimb *
  • scikit-optimize *
  • scipy *
  • sphinx >=4.0
  • sphinx-copybutton *
  • sphinx-intl *
  • tensorflow ==2.7
  • tensorflow_quantum ==0.6.1
  • tensornetwork ==0.4.6
  • torch *
  • torchvision *
requirements/requirements-extra.txt pypi
  • qiskit *
  • torch *
requirements/requirements-rtd.txt pypi
  • cirq *
  • furo ==2022.4.7
  • graphviz *
  • ipykernel *
  • jax *
  • nbsphinx *
  • networkx *
  • numpy *
  • optax *
  • scipy *
  • sphinx ==4.3.2
  • sphinx-copybutton *
  • tensorflow *
  • tensornetwork *
  • torch *
requirements/requirements.txt pypi
  • cirq ==0.11
  • graphviz *
  • jax *
  • jaxlib *
  • networkx *
  • numpy *
  • optax *
  • scipy *
  • tensorflow ==2.7
  • tensornetwork *
setup.py pypi
  • numpy *
docker/Dockerfile docker
  • nvidia/cuda 11.0.3-cudnn8-runtime-ubuntu18.04 build
.github/workflows/ci.yml actions
  • actions/checkout v2 composite
  • actions/setup-python v4 composite
  • codecov/codecov-action v2 composite
.github/workflows/nightly_release.yml actions
  • actions/checkout v2 composite
  • actions/setup-python v4 composite
  • pypa/gh-action-pypi-publish release/v1 composite
.devcontainer/Dockerfile docker
  • mcr.microsoft.com/vscode/devcontainers/python 0-${VARIANT} build
requirements/requirements-types.txt pypi
  • types-docutils *
  • types-protobuf *
  • types-pytz *
  • types-requests *
  • types-setuptools *
  • types-toml *
  • types-typed-ast *
  • types-urllib3 *