arXiv · 2503.14422
QSTToolkit: A Python Library for Deep Learning Powered Quantum State Tomography
Abstract
We introduce QSTToolkit, a Python library for performing quantum state tomography (QST) on optical quantum state measurement data. The toolkit integrates traditional Maximum Likelihood Estimation (MLE) with deep learning-based techniques to reconstruct quantum states. It includes comprehensive noise models to simulate both intrinsic state noise and measurement imperfections, enabling the realistic recreation of experimental data. QSTToolkit bridges TensorFlow, a leading deep learning framework, with QuTiP, a widely used quantum physics toolbox for Python. This paper describes the library's features, including its data generation capabilities and the various QST methods implemented. QSTToolkit is available at https://pypi.org/project/qsttoolkit/, with full documentation at https://qsttoolkit.readthedocs.io/en/latest/.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
George FitzGerald, Will Yeadon. 2025-03-18. QSTToolkit: A Python Library for Deep Learning Powered Quantum State Tomography. https://arxiv.org/abs/2503.14422
Cite the original work for its findings. Save a collection to share your selection of sources.