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 60+ example scripts and 30+ tutorial notebooks. API docstrings and test cases in tests are also informative.

The following are some minimal demos.

• Circuit manipulation:

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:

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

• Automatic differentiations with jit:

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.array_to_tensor(1.0)
print(g(theta))

Install

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

pip install tensorcircuit

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

pip install tensorcircuit[tensorflow]

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

For the nightly build of tensorcircuit with new features, try:

pip uninstall tensorcircuit
pip install tensorcircuit-nightly

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 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.

  • 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 frameoworks all at the same time.

Contributing

Status

This project is released by Tencent Quantum Lab and 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 lab and the open source community.

Citation

If this project helps in your research, please cite our software whitepaper published in Quantum:

TensorCircuit: a Quantum Software Framework for the NISQ Era

which is also a good introduction to the software.

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 💻 📖 💡 🤔 🚇 🚧 🔬 👀 🌍 ⚠️ ✅ 📢 💬	Yuqin Chen 💻 📖 💡 🤔 🔬 ⚠️ ✅ 📢	Jiezhong Qiu 💻 💡 🤔 🔬	Weitang Li 💻 📖 🤔 🔬 ⚠️ 📢	Jiace Sun 💻 📖 💡 🤔 🔬 ⚠️	Zhouquan Wan 💻 📖 💡 🤔 🔬 ⚠️ ✅
Shuo Liu 💡 🔬 ✅	Hao Yu 💻 📖 🚇 ⚠️ ✅	Xinghan Yang 📖 🌍 ✅	JachyMeow ✅ 🌍	Zhaofeng Ye 🎨	erertertet 💻 📖 ⚠️
Yicong Zheng ✅	Zixuan Song 📖 🌍 💻 ⚠️	Hao Xie 📖	Pramit Singh ⚠️	Jonathan Allcock 📖 🤔 📢	nealchen2003 📖
隐公观鱼 💻 ⚠️	WiuYuan 💡	Felix Xu ✅

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.

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.

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.

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.