Skip to main content

Lightweight Quantum Simulation Suite with VQE and QPE modules (PennyLane-based)

Project description

Quantum Simulation Suite — VQE + QPE (PennyLane)

PyPI Version PyPI Downloads Tests Python Versions License

A modern, modular, and fully reproducible quantum-chemistry simulation suite built on
PennyLane, featuring:

  • Variational Quantum Eigensolver (VQE)
  • State-Specific VQE (SSVQE)
  • Quantum Phase Estimation (QPE)
  • Unified molecule registry, geometry generators, and plotting tools
  • Consistent caching and reproducibility across all solvers

This project refactors all previous notebooks into a clean Python package with
a shared vqe_qpe_common/ layer for Hamiltonians, molecules, geometry, and plotting.

  • THEORY.md — Full background on VQE, QPE, ansatzes, optimizers, mappings, and noise models.
  • USAGE.md — Practical guide for installation, CLI usage, running simulations, caching, and plotting.

These documents complement the README and provide both the theoretical foundation and the hands-on execution details of the VQE/QPE suite.


Project Structure

Variational_Quantum_Eigensolver/
├── README.md
├── THEORY.md
├── USAGE.md
├── LICENSE
├── pyproject.toml
│
├── vqe_qpe_common/          # Shared logic for VQE + QPE
│   ├── molecules.py         # Unified molecule registry
│   ├── geometry.py          # Bond/angle geometry generators
│   ├── hamiltonian.py       # Unified Hamiltonian builder (PennyLane/OpenFermion)
│   └── plotting.py          # Shared plotting + filename builders
│
├── vqe/                     # Variational Quantum Eigensolver package
│   ├── __main__.py          # CLI: python -m vqe
│   ├── core.py              # VQE orchestration (runs, scans, sweeps)
│   ├── engine.py            # Devices, noise, ansatz/optimizer plumbing
│   ├── ansatz.py            # UCCSD, RY-CZ, HEA, minimal ansätze
│   ├── optimizer.py         # Adam, GD, Momentum, SPSA, etc.
│   ├── hamiltonian.py       # VQE-specific wrapper → uses vqe_qpe_common.hamiltonian
│   ├── io_utils.py          # JSON caching, run signatures
│   ├── visualize.py         # Convergence, scans, noise plots
│   └── ssvqe.py             # Subspace-search VQE (excited states)
│
├── qpe/                     # Quantum Phase Estimation package
│   ├── __main__.py          # CLI: python -m qpe
│   ├── core.py              # Controlled-U, trotterized dynamics, iQFT
│   ├── hamiltonian.py       # QPE-specific wrapper → uses vqe_qpe_common.hamiltonian
│   ├── io_utils.py          # JSON caching, run signatures
│   ├── noise.py             # Depolarizing + amplitude damping channels
│   └── visualize.py         # Phase histograms + sweep plots
│
├── results/                 # JSON outputs
├── images/                  # Saved plots (VQE + QPE)
├── data/                    # Optional molecule configs, external data
│
└── notebooks/               # Notebooks importing from the vqe/ and qpe/ packages

This structure ensures:

  • VQE and QPE share the same chemistry (vqe_qpe_common/)
  • All results are cached consistently (results/)
  • All plots use one naming system (vqe_qpe_common/plotting.py)
  • Notebooks import from the real package (no duplicated code)
  • CLI tools are production-ready (python -m vqe, python -m qpe)

⚙️ Installation

Install from PyPI

pip install vqe-pennylane

Install from source (development mode)

git clone https://github.com/SidRichardsQuantum/Variational_Quantum_Eigensolver.git
cd Variational_Quantum_Eigensolver
pip install -e .

Confirm installation

python -c "import vqe, qpe; print('VQE+QPE imported successfully!')"

Common Core (Shared by VQE & QPE)

The following modules ensure full consistency between solvers:

Module Purpose
vqe_qpe_common/molecules.py Canonical molecule definitions
vqe_qpe_common/geometry.py Bond/angle/coordinate generators
vqe_qpe_common/hamiltonian.py Hamiltonian construction + OpenFermion fallback
vqe_qpe_common/plotting.py Unified filename builder + PNG export

🔹 VQE Package

Features:

  • Ground-state VQE
  • Excited-state SSVQE
  • Geometry scans
  • Noise sweeps
  • Mapping comparisons
  • Optimizer registry
  • Result caching

Run example:

from vqe.core import run_vqe
result = run_vqe("H2", ansatz_name="UCCSD", optimizer_name="Adam", n_steps=50)
print(result["energy"])

🔹 QPE Package

Features:

  • Noiseless & noisy QPE
  • Trotterized exp(-iHt)
  • Inverse QFT
  • Noise channels
  • Cached results

Example:

from vqe_qpe_common.hamiltonian import build_hamiltonian
from qpe.core import run_qpe

H, n_qubits, hf_state = build_hamiltonian(["H","H"], coords, 0, "STO-3G")
result = run_qpe(H, hf_state, n_ancilla=4)

CLI Usage

VQE

python -m vqe -m H2 -a UCCSD -o Adam --steps 50

QPE

python -m qpe --molecule H2 --ancillas 4 --shots 2000

🧪 Tests

pytest -v

📘 Author: Sid Richards (SidRichardsQuantum)

LinkedIn: https://www.linkedin.com/in/sid-richards-21374b30b/

This project is licensed under the MIT License - see the LICENSE file for details.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

vqe_pennylane-0.2.2.tar.gz (55.5 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

vqe_pennylane-0.2.2-py3-none-any.whl (50.1 kB view details)

Uploaded Python 3

File details

Details for the file vqe_pennylane-0.2.2.tar.gz.

File metadata

  • Download URL: vqe_pennylane-0.2.2.tar.gz
  • Upload date:
  • Size: 55.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.1

File hashes

Hashes for vqe_pennylane-0.2.2.tar.gz
Algorithm Hash digest
SHA256 b33fe3ecbe16299424b2b701bac413d15b653b205ef75d025d098673827e036b
MD5 a6a3652ff6ecd7404029ed126e6f9186
BLAKE2b-256 289fc576d08102ffd5ba42477cc9bac525c0482883e3c36dfc9354359b661151

See more details on using hashes here.

File details

Details for the file vqe_pennylane-0.2.2-py3-none-any.whl.

File metadata

  • Download URL: vqe_pennylane-0.2.2-py3-none-any.whl
  • Upload date:
  • Size: 50.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.1

File hashes

Hashes for vqe_pennylane-0.2.2-py3-none-any.whl
Algorithm Hash digest
SHA256 13263cfd1c89a6c8a68bd76fe3fa0aab2cd4d77ba507af8e208a5aab69418e5c
MD5 91fd52d839ab81c7e3b6bf339e3270f2
BLAKE2b-256 e8a0bd76ba31a589be6107ed9e1d4c22b2ba594da870976d6e1613da98fc4ca8

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page