Skip to content

Repository files navigation

Quantum Pipeline

Repository: GitHub (primary) · Codeberg (mirror)

PyPI version Total Downloads PyPI - Downloads Docker Pulls

Documentation · Quick Start · Examples


A framework for running quantum algorithms, with optional Kafka streaming, Spark processing, and Airflow orchestration. Currently implements the Variational Quantum Eigensolver (VQE) for ground-state energy estimation. Built as a Bachelor of Engineering thesis project and continued as a Master of Engineering thesis project.

Quick Start

pip install quantum-pipeline
quantum-pipeline -f molecules.json -b sto3g --max-iterations 100 --optimizer L-BFGS-B

Or with Docker:

docker pull straightchlorine/quantum-pipeline:cpu
docker run --rm straightchlorine/quantum-pipeline:cpu -f data/molecules.json -b sto3g --max-iterations 100

See the installation guide for detailed setup, including GPU acceleration and full platform deployment.

Features

Quantum Computing - VQE execution with multiple optimizers (L-BFGS-B, COBYLA, SLSQP, and others), configurable ansatz circuits (EfficientSU2, RealAmplitudes, ExcitationPreserving), parameter initialization strategies (random or Hartree-Fock based), multiple basis sets (sto-3g, 6-31g, cc-pVDZ), and GPU acceleration via CUDA. Learn more

Data Platform - Kafka streaming with Avro serialization, Spark-based ML feature engineering, Airflow workflow orchestration. Architecture overview

ML Pipeline - Work in progress: convergence prediction and energy estimation models trained on VQE experiment data. Includes dedicated Docker Compose stack for this purpose.

Monitoring - Prometheus metrics export, Grafana dashboard and resource tracking. Configurable via environment variables or CLI flags. Monitoring setup

Deployment - Docker images for CPU and GPU (quantum-pipeline:cpu, quantum-pipeline:gpu). The GPU image on Docker Hub is built for Ampere (RTX 30xx) - rebuild with CUDA_ARCH=6.1 for Pascal or 8.9 for Ada Lovelace. Deployment guide

Python API

from quantum_pipeline.runners.vqe_runner import VQERunner

runner = VQERunner(
    filepath='data/molecules.json',
    basis_set='sto3g',
    max_iterations=100,
    optimizer='COBYLA',
)
runner.run()

See the API documentation for all constructor parameters.

License

MIT License. See LICENSE for details.

Contact

About

A Python module for executing and monitoring quantum algorithms across local simulators and IBM Quantum platforms. Seamlessly handles data collection, organization, and streaming to Apache Kafka.

Topics

Resources

Stars

11 stars

Watchers

1 watching

Forks

Releases

Packages

Used by

Contributors

Languages