Krotov Python Package

Source code on Github Documentation Krotov on the Python Package Index Docs Tests Codecov BSD License Launch Binder DOI

Python implementation of Krotov’s method for quantum optimal control.

This implementation follows the original implementation in the QDYN Fortran library.

The krotov package is built on top of QuTiP.

Development happens on Github. You can read the full documentation online.

If you use the krotov package in your research, please cite it.

Purpose

Optimal control is a cornerstone of quantum technology: relying not just on a passive understanding of quantum mechanics, but on the active utilization of the quantum properties of matter. Quantum optimal control asks how to manipulate the dynamics of a quantum system in some desired way. This is essential for the realization of quantum computers and related technologies such as quantum sensing.

Krotov’s method and GRAPE are the two leading gradient-based optimization algorithms used in numerical quantum optimal control. Krotov’s method distinguishes itself by guaranteeing monotonic convergence for near-continuous control fields. This makes is particularly useful for exploring the limits of controllability in a physical system. While GRAPE is found in various software packages, there has not been an open source implementation of Krotov’s method to date. Our package provides that missing implementation.

The Krotov package targets both students wishing to enter the field of quantum control and researchers in the field. It was designed towards the following goals:

  • Leverage the QuTiP library as a platform for numerically describing quantum systems.

  • Provide a collection of examples inspired by recent publications in the Jupyter notebook format, allowing for interactive exploration of the method.

  • Define a general interface for formulating any quantum control problem, which may extend to other optimization methods in the future.

  • Serve as a reference implementation of Krotov’s method, and as a foundation against which to test other implementations.

  • Enable the more widespread use of Krotov’s method, for example in the design of experiments.

Prerequisites

The Krotov package is available for Python versions >= 3.7. Its main dependency is QuTiP (apart from the core packages of the Python scientific ecosystem). Thus, you should consider QuTiP’s installation instructions.

In any case, using some sort of virtual environment is strongly encouraged. Most packages in the Python scientific ecosystem are now available as wheels, making installation via pip easy. If you need to install packages that do not provide compiled whels, and if you do not have the necessary compilers installed (Windows, macOS), the conda package manager provides a good solution.

Installation

To install the latest released version of krotov into your current (conda) environment, run this command in your terminal:

python -m pip install krotov

This is the preferred method to install the krotov package, as it will always install the most recent stable release.

If you don’t have pip installed, the Python installation guide, respectively the Python Packaging User Guide can guide you through the process.

To install the latest development version of krotov from Github:

python -m pip install git+https://github.com/qucontrol/krotov.git@master#egg=krotov

Usage

To use Krotov’s method for quantum optimal control in a Python script or Jupyter notebook, start with:

import krotov
import qutip

Then,

  1. define the necessary quantum operators and states using QuTiP.

  2. create a list of objectives, as instances of krotov.Objective.

  3. call krotov.optimize_pulses to perform an optimization of an arbitrary number of control fields over all the objectives.

See Using Krotov with QuTiP and Examples for details.

Citing the Krotov Package

Attention

Please cite the krotov package as

  • M. H. Goerz et al., Krotov: A Python implementation of Krotov’s method for quantum optimal control, SciPost Phys. 7, 080 (2019)

You can also print this from krotov.__citation__:

>>> print(krotov.__citation__)
M. H. Goerz et al., Krotov: A Python implementation of Krotov's method for quantum optimal control, SciPost Phys. 7, 080 (2019)

The corresponding BibTeX entry is available in krotov.__bibtex__:

>>> print(krotov.__bibtex__)
@article{GoerzSPP2019,
    author = {Michael H. Goerz and Daniel Basilewitsch and Fernando Gago-Encinas and Matthias G. Krauss and Karl P. Horn and Daniel M. Reich and Christiane P. Koch},
    title = {Krotov: A {Python} implementation of {Krotov's} method for quantum optimal control},
    journal={SciPost Phys.},
    volume={7},
    pages={80},
    year={2019},
    doi={10.21468/SciPostPhys.7.6.080},
}