View on GitHub
pyGSTi icon

pyGSTi

A python implementation of Gate Set Tomography

View the online documentation Download this project as a .zip file Download this project as a tar.gz file

What is pyGSTi?

pyGSTi is an open-source software for modeling and characterizing noisy quantum information processors (QIPs), i.e., systems of one or more qubits. It is licensed under the Apache License, Version 2.0. Copyright information can be found in NOTICE, and the license itself in LICENSE.

There are three main objects in pyGSTi:

You can do various things by with these objects:

In particular, there are a number of characterization protocols currently implemented in pyGSTi:

PyGSTi is designed with a modular structure so as to be highly customizable and easily integrated to new or existing python software. It runs using python2.7 or python3. To faclilitate integration with software for running cloud-QIP experiments, pyGSTi Circuit objects can be converted to IBM's OpenQASM and Rigetti Quantum Computing's Quil circuit description languages.

Installation

Apart from several optional Cython modules, pyGSTi is written entirely in Python. To install pyGSTi and only its required dependencies run:

pip install pygsti

Or, to install pyGSTi with all its optional dependencies too, run:

pip install pygsti[complete]

The disadvantage to these approaches is that the numerous tutorials included in the package will then be buried within your Python's site_packages directory, which you'll likely want to access later on. Alternatively, you can locally install pyGSTi using the following commands:

cd <install_directory>
git clone https://github.com/pyGSTio/pyGSTi.git
cd pyGSTi
pip install -e .[complete]

As above, you can leave off the .[complete] if you only went the minimal set of dependencies installed. You could also replace the git clone ... command with unzip pygsti-0.9.x.zip where the latter file is a downloaded pyGSTi source archive. Any of the above installations should build the set of optional Cython extension modules if a working C/C++ compiler and the Cython package are present. If, however, compilation fails or you later decided to add Cython support, you can rebuild the extension modules (without reinstalling) if you've followed the local installation approach above using the command:

python setup.py build_ext --inplace

Finally, Jupyter notebook is highly recommended as it is generally convenient and the format of the included tutorials and examples. It is installed automatically when [complete] is used, otherwise it can be installed separately.

Getting Started

Here's a couple of simple examples to get you started.

Circuit simulation

To compute the outcome probabilities of a circuit, you just need to create a Circuit object (describing your circuit) and a Model object containing the operations contained in your circuit. Here we use a "stock" single-qubit Model containing Idle, X(π/2), and Y(π/2) gates labelled Gi, Gx, and Gy, respectively:

import pygsti
from pygsti.construction import std1Q_XYI

mycircuit = pygsti.obj.Circuit( ('Gx','Gy','Gx') )
model = std1Q_XYI.target_model()
outcome_probabilities = model.probs(mycircuit)

Gate Set Tomography

Gate Set Tomography is used to characterize the operations performed by hardware designed to implement a (small) system of quantum bits (qubits). Here's the basic idea:

  1. you tell pyGSTi what gates you'd ideally like to perform
  2. pyGSTi tells you what circuits it want's data for
  3. you perform the requested experiments and place the resulting data (outcome counts) into a text file that looks something like:

    ## Columns = 0 count, 1 count
    {} 0 100  # the empty sequence (just prep then measure)
    Gx 10 90  # prep, do a X(pi/2) gate, then measure
    GxGy 40 60  # prep, do a X(pi/2) gate followed by a Y(pi/2), then measure
    Gx^4 20 80  # etc...
    
  4. pyGSTi takes the data file and outputs a "report" - currently a HTML web page.

In code, running GST looks something like this:

import pygsti
from pygsti.construction import std1Q_XYI

# 1) get the ideal "target" Model (a "stock" model in this case)
mdl_ideal = std1Q_XYI.target_model()

# 2) get the building blocks needed to specify which circuits are needed
prepfids, measfids, germs = std1Q_XYI.prepStrs, std1Q_XYI.effectStrs, std1Q_XYI.germs
maxLengths = [1,2,4] # user-defined: how long do you want the circuits?

# 3) generate a list of circuits for GST & write a data-set template
listOfExperiments = pygsti.construction.make_lsgst_experiment_list(
    mdl_ideal, prepfids, measfids, germs, maxLengths)
pygsti.io.write_empty_dataset("MyData.txt", listOfExperiments, "## Columns = 0 count, 1 count")

# STOP! "MyData.txt" now has columns of zeros where actual data should go.
# REPLACE THE ZEROS WITH ACTUAL DATA, then proceed with:
ds = pygsti.io.load_dataset("MyData.txt") # load data -> DataSet object

# 4) run GST
results = pygsti.do_stdpractice_gst(ds, mdl_ideal, prepfids, measfids, germs, maxLengths)

# 5) Create a nice HTML report detailing the results
pygsti.report.create_standard_report(results, filename="myReport", title="Sample Report")

Tutorials and Examples

There are numerous tutorials (meant to be pedagogical) and examples (meant to be demonstrate how to do some particular thing) in the form of Jupyter notebooks beneath the pyGSTi/jupyter_notebooks directory. The root START HERE notebook will direct you where to go based on what you're most interested in learning about. You can view the read-only GitHub version of this notebook or you can explore the tutorials interactively using JupyterHub via Binder. Note the existence of a FAQ, which addresses common issues.

Running notebooks locally

While it's possible to view the notebooks on GitHub using the links above, it's usually nicer to run them locally so you can mess around with the code as you step through it. To do this, you'll need to start up a Jupyter notebook server using the following steps (this assumes you've followed the local installation directions above):

The Jupyter server should open up your web browser to the server root, from where you can start the first "START_HERE.ipynb" notebook. Note that the key command to execute a cell within the Jupyter notebook is Shift+Enter, not just Enter.

Documentation

Detailed documentation of the pyGSTi code (automatically generated from the source code's docstrings) can be found here. These pages provide a searchable reference for just about everything pyGSTi can do. Please let us know if you find any errors.

License

pyGSTi is free and open-source software: you can redistribute it and/or modify it under the terms of the Apache License Version 2.0. To cite this work, please use the DOI: 10.5281/zenodo.46200

Questions?

For help and support with pyGSTi, please contact the authors at pygsti@sandia.gov.