quantum-lattice-models 0.1.2

Small dense quantum lattice Hamiltonians for exact diagonalization and quantum algorithm prototypes.

Quantum Lattice Models

Quantum Lattice Models is a lightweight, package-first Python library for constructing, analyzing, plotting, and exporting small lattice Hamiltonians used in physics workflows and quantum algorithm research prototypes.

PyPI: https://pypi.org/project/quantum-lattice-models/

This repository is organized as an installable package first. The real logic lives in src/quantum_lattice_models/; notebooks, scripts, and examples should stay thin and import the public package API. quantum_lattice_models.models remains the compatibility import surface, while implementations are split across focused modules such as spin, tight_binding, hubbard, and topological.

Implemented Models

• Transverse-field Ising spin chain
• Longitudinal-field Ising spin chain
• Next-nearest-neighbor Ising spin chain
• Anisotropic Heisenberg spin chain
• XY spin chain
• XXZ spin chain
• Frustrated J1-J2 Heisenberg spin chain
• Two-leg Heisenberg spin ladder
• Truncated Bose-Hubbard chain
• Spinful Fermi-Hubbard chain
• Kitaev-chain Bogoliubov-de Gennes matrix
• Su-Schrieffer-Heeger single-particle tight-binding model
• Rice-Mele single-particle chain
• Generic one-dimensional single-particle tight-binding chain
• Square-lattice single-particle tight-binding model
• Harper-Hofstadter square-lattice model
• Aubry-Andre-Harper quasiperiodic tight-binding chain
• Haldane honeycomb-lattice model
• Triangular-lattice single-particle tight-binding model
• Kagome-lattice single-particle tight-binding model
• User-defined graph/lattice tight-binding models

Spin-chain Hamiltonians are dense qubit-space matrices Tight-binding Hamiltonians are single-particle matrices. This distinction is intentional and explicit.

Why Lattice Models Matter

Small lattice Hamiltonians are useful because they are concrete, inspectable testbeds. They connect physics intuition to numerical linear algebra, and they give quantum algorithm researchers controlled problems for VQE, QPE, QSVT, spectral transforms, quantum walks, and simulation workflows.

This package does not claim quantum advantage. It provides honest small-system tools for exact diagonalization, prototyping, teaching, and notebook-first experiments.

Installation

From a local checkout:

python -m venv .venv
source .venv/bin/activate
python -m pip install --upgrade pip
python -m pip install -e ".[dev]"

Minimal runtime dependencies are numpy, scipy, and matplotlib.

PennyLane export is optional:

pip install -e ".[pennylane]"

Notebook support is optional:

python -m pip install -e ".[notebooks]"
python -m ipykernel install --user --name quantum-lattice-models --display-name "Quantum Lattice Models"

Quickstart

from quantum_lattice_models.models import transverse_field_ising
from quantum_lattice_models.spectra import ground_energy, spectral_gap

H = transverse_field_ising(n_sites=4, j=1.0, h=0.5, periodic=False)

print(H.shape)
print(ground_energy(H))
print(spectral_gap(H))

from quantum_lattice_models.models import ssh_model, ssh_edge_state_localizations
from quantum_lattice_models.spectra import eigensystem

H = ssh_model(n_cells=8, t1=0.5, t2=1.0, periodic=False)
values, vectors = eigensystem(H)
weights = ssh_edge_state_localizations(vectors, n_cells=8, edge_cells=2)

User-Defined Lattices

Users can build custom single-particle tight-binding models without adding a new function to the package:

from quantum_lattice_models import Lattice, TightBindingModel

lattice = Lattice(
    positions=[(0.0, 0.0), (1.0, 0.0), (0.5, 0.8)],
    bonds=[(0, 1), (1, 2, 0.25j), (2, 0)],
)

H = TightBindingModel(lattice).hamiltonian(hopping=1.0, onsite=[0.0, 0.1, 0.0])

Two-item bonds use -hopping as the matrix element. Three-item bonds use the third value directly, which allows complex hoppings and Peierls phases.

Custom builders can also be registered for discovery in notebooks and the CLI:

from quantum_lattice_models.registry import register_model

register_model(
    "my_model",
    category="user",
    basis="single particle",
    dimension="n_sites",
    return_type="LatticeHamiltonian",
    description="My custom tight-binding model",
    builder=my_builder,
    defaults={"n_sites": 4},
)

The plotting helpers can use lattice position metadata directly:

from quantum_lattice_models.plotting import plot_lattice_graph, plot_lattice_state
from quantum_lattice_models.spectra import eigensystem

plot_lattice_graph(H, show_colorbar=True)
values, vectors = eigensystem(H)
plot_lattice_state(H, vectors[:, 0])

Repository Structure

src/quantum_lattice_models/  Package source
tests/                       Pytest test suite
examples/                    Command-line examples that save plots
notebooks/                   Thin-client exploratory notebooks
README.md                    Project overview
CHANGELOG.md                 Release notes
USAGE.md                     API examples
THEORY.md                    Model and method notes
RESULTS.md                   Generated results

Key package modules:

spin.py                      Dense spin-chain and ladder builders
tight_binding.py             Single-particle tight-binding builders
hubbard.py                   Bose-Hubbard and Fermi-Hubbard builders
topological.py               Haldane, Hofstadter, and Kitaev builders
geometry.py                  Coordinate helpers for plotting
lattice.py                   User-defined lattice containers and custom builders
registry.py                  Structured model metadata
cli.py                       quantum-lattice command-line entry point
models.py                    Backwards-compatible re-export layer

Notebooks as Thin Clients

Notebooks should import from quantum_lattice_models rather than defining their own model logic. A notebook can choose parameters, run spectra, plot results, and tell a story. The package should remain the source of truth.

Current notebooks:

• notebooks/ising_spin_chains.ipynb
• notebooks/ssh_rice_mele_comparison.ipynb
• notebooks/hofstadter_flux_sweep.ipynb
• notebooks/hubbard_exact_diagonalization.ipynb
• notebooks/haldane_kagome_lattices.ipynb
• notebooks/model_registry_and_cli.ipynb
• notebooks/kitaev_bdg_symmetry.ipynb
• notebooks/heisenberg_ladder_spectrum.ipynb
• notebooks/sparse_dense_scaling.ipynb
• notebooks/cli_plot_walkthrough.ipynb

Development

Use the virtual environment for examples, notebooks, tests, and packaging commands. The standard local checks are:

make format
make lint
make test

The Makefile runs Black one file at a time to avoid multi-file formatter stalls observed in some Codespace environments. Before a release, also run Ruff across the full repository so notebook code cells are checked:

python -m ruff check .
python -m pytest -q

Install .[pennylane] before the test command to exercise the optional PennyLane export test instead of skipping it.

Publishing a release

Pushing a tag such as v0.1.2 runs .github/workflows/publish.yml. The workflow verifies that the tag matches the version in pyproject.toml, runs the release checks, builds the wheel and source distribution, and publishes them to PyPI.

Configure a PyPI Trusted Publisher for this repository before pushing the first release tag:

• Owner: SidRichardsQuantum
• Repository: Quantum_Lattice_Models
• Workflow filename: publish.yml
• Environment name: pypi

No PyPI API token is required. The GitHub pypi environment can optionally require manual approval before publication.

Limitations / Truth Contract

• Dense spin-chain matrices scale as 2**n_sites by 2**n_sites.
• These tools are for small systems, education, exact diagonalization, and research prototypes.
• SSH, Rice-Mele, square, Harper-Hofstadter, Haldane, triangular, kagome, and generic tight-binding builders return single-particle matrices, not many-body Fock-space Hamiltonians.
• The Bose-Hubbard builder uses a truncated local occupation basis.
• The Fermi-Hubbard builder uses a dense occupation-number basis with explicit fermionic signs.
• The Kitaev-chain builder returns a Bogoliubov-de Gennes matrix, not a many-body Hamiltonian.
• Sparse builders are available for selected tight-binding and Hubbard chains, but exact diagonalization remains a small-system workflow.
• PennyLane is optional and only used when explicitly installed.
• The project is a backend for experiments, not a benchmark suite proving speedup or quantum advantage.

---

Support development

If this repository is useful for research, learning, or experimentation, you can support continued development via GitHub Sponsors:

https://github.com/sponsors/SidRichardsQuantum

Sponsorship helps support ongoing work on open-source implementations of quantum algorithms, including improvements to documentation, reproducible workflows, and example notebooks.

Support helps maintain and expand practical tooling for variational quantum methods, quantum simulation workflows, and related experimentation.

Citation

Sid Richards (2026)

Unified Variational and Phase-Estimation Quantum Simulation Suite

Author

Sid Richards

• LinkedIn: sid-richards-21374b30b
• GitHub: SidRichardsQuantum

License

MIT. See LICENSE.