PtyLab.py: Unified Ptychography Toolbox

Python 3.10+ Tests

Getting Started | Installation | Development | Documentation

PtyLab is an inverse modeling toolbox for Conventional (CP) and Fourier (FP) ptychography in a unified framework. For more information please check the paper.

Key Features

Classic engines: ePIE, mPIE, mqNewton, qNewton
Advanced corrections: position correction (pcPIE), defocus correction (zPIE), angle correction (aPIE), orthogonal probe relaxation (OPR)
Multi-modal: multi-slice, multi-wavelength, mixed-state object and probe
Multiple propagators: Fraunhofer, Fresnel, Angular Spectrum (ASP), scaled ASP, polychromatic variants
GPU acceleration: same code runs on CPU and GPU

The reconstructed output is a 6D array of shape (nlambda, nosm, npsm, nslice, No, No):

Dim	Meaning
`nlambda`	wavelengths
`nosm`	object state mixture
`npsm`	probe state mixture
`nslice`	depth slices
`No`	output frame size

Getting started

Try the demo in Google Colab:

demo

For more use cases, start with the tutorials in our documentation, then see the example_scripts directory.

Installation

Install from PyPI within your virtual environment:

pip install ptylab

For much faster installs, we recommend uv: uv pip install ptylab

This package uses cupy to utilize GPU for faster reconstruction. To enable GPU support:

pip install "ptylab[gpu]"

You can check if GPU is detected with ptylab check gpu.

For the latest unreleased changes on main:

pip install git+https://github.com/PtyLab/PtyLab.py.git

Development

Clone the repo and install dev and gpu dependencies with uv:

git clone git@github.com:PtyLab/PtyLab.py.git
cd PtyLab.py
uv sync --extra dev,gpu # remove the GPU flag if you are on CPU

This creates a .venv virtual environment in the project root. Select this environment from your IDE.

Add tests for new implementations and run the suite:

uv run pytest tests

Citation

If you use this package, please cite:

@article{Loetgering:23,
        author = {Lars Loetgering and Mengqi Du and Dirk Boonzajer Flaes and Tomas Aidukas and Felix Wechsler and Daniel S. Penagos Molina and Max Rose and Antonios Pelekanidis and Wilhelm Eschen and J\"{u}rgen Hess and Thomas Wilhein and Rainer Heintzmann and Jan Rothhardt and Stefan Witte},
        journal = {Opt. Express},
        number = {9},
        pages = {13763--13797},
        publisher = {Optica Publishing Group},
        title = {PtyLab.m/py/jl: a cross-platform, open-source inverse modeling toolbox for conventional and Fourier ptychography},
        volume = {31},
        month = {Apr},
        year = {2023},
        doi = {10.1364/OE.485370},
}

Download files

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

Source Distribution

ptylab-0.2.6.tar.gz (92.7 kB view details)

Uploaded Aug 3, 2026 Source

Built Distribution

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

The dropdown lists show the available interpreters, ABIs, and platforms. Enable javascript to be able to filter the list of wheel files.

ptylab-0.2.6-py3-none-any.whl (122.3 kB view details)

Uploaded Aug 3, 2026 Python 3

File details

Details for the file ptylab-0.2.6.tar.gz.

File metadata

Download URL: ptylab-0.2.6.tar.gz
Upload date: Aug 3, 2026
Size: 92.7 kB
Tags: Source
Uploaded using Trusted Publishing? Yes
Uploaded via: twine/6.1.0 CPython/3.13.14

File hashes

Hashes for ptylab-0.2.6.tar.gz
Algorithm	Hash digest
SHA256	`9ec37a9db9b347de61b2c767c230f39c04e1524080224ebae9514cc220474261`
MD5	`a8dc89a7cbe3571ce0881ed88a509ba3`
BLAKE2b-256	`db324f72454373f1f6c49dbf3977e919cf73c283647ffab96fc65c3391ed7d5c`

See more details on using hashes here.

Provenance

The following attestation bundles were made for ptylab-0.2.6.tar.gz:

Publisher: publish.yml on PtyLab/PtyLab.py

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Statement:
- Statement type: https://in-toto.io/Statement/v1
- Predicate type: https://docs.pypi.org/attestations/publish/v1
- Subject name: ptylab-0.2.6.tar.gz
- Subject digest: 9ec37a9db9b347de61b2c767c230f39c04e1524080224ebae9514cc220474261
- Sigstore transparency entry: 2334319738
- Sigstore integration time: Aug 3, 2026
Source repository:
- Permalink: PtyLab/PtyLab.py@606195d3b1a52765540d098e2b06a9d1abdc045d
- Branch / Tag: refs/tags/v0.2.6
- Owner: https://github.com/PtyLab
- Access: public
Publication detail:
- Token Issuer: https://token.actions.githubusercontent.com
- Runner Environment: github-hosted
- Publication workflow: publish.yml@606195d3b1a52765540d098e2b06a9d1abdc045d
- Trigger Event: release