Pypomp
Pypomp is a Python/JAX library for modeling and inference using partially observed Markov process (POMP) models, also called state-space models or hidden Markov models.
🚀 Quick Links
| 📖 Read the Documentation | 🎓 Introductory Tutorial |
|---|---|
| 🏫 SBIED Course (Practical Modeling) | 📊 Quantitative Benchmarks |
Installation
Install Pypomp using pip:
pip install pypomp
📝 Note: Pypomp is powered by JAX for GPU acceleration and just-in-time (JIT) compilation. To configure JAX for GPU computing, please follow the JAX Installation Guide.
Quick Start
Get started quickly by running a particle filter on a built-in Linear Gaussian model:
import jax
import pypomp as pp
# 1. Initialize a built-in Linear Gaussian model with 50 time steps
model = pp.models.lg(T=50)
# 2. Run the particle filter with J=1000 particles
key = jax.random.key(1)
model.pfilter(J=1000, key=key)
# 3. Retrieve and print the estimated log-likelihood using .results()
results_df = model.results()
log_lik = results_df["logLik"].iloc[0]
print(f"Log-Likelihood: {log_lik:.4f}")
Expected Users
-
Scientists wanting to perform data analysis on a dynamic system via a POMP model.
-
Researchers wishing to develop novel inference methodology. Pypomp provides an abstract representation of POMP models that enables researchers to develop, test, and deploy novel algorithms applicable to arbitrary nonlinear non-Gaussian POMP models.
-
Researchers familiar with the pomp R package. Pypomp extends R-pomp by supporting GPU computing, automatic differentiation, and just-in-time compilation. Conceptually, Pypomp is similar to R-pomp, and so case studies listed in the R-pomp package bibliography are pertinent.
Key Features
-
State & Parameter Inference: Parameter estimation, model evaluation, and latent state estimation for nonlinear, non-Gaussian POMP models via the particle filter.
-
Differentiable Particle Filtering: Gradient descent using a new particle filter gradient estimate. This provides state-of-the-art simulation-based maximum likelihood and Bayesian inference.
-
JAX-Backed Performance: Leverages JAX to provide GPU support, automatic differentiation, and just-in-time compilation.
Governance and Contributions
The Pypomp library is run by the Pypomp organization.
All contributions are welcome. Please review our Contributing Guide before submitting pull requests or raising issues.
For governance details, you can contact the core development team or refer to the pypomp governance profile.
📚 Citation
If you use Pypomp in your research, please cite it as:
@software{pypomp2024github,
author = {Aaron Abkemeier and Jun Chen and Edward Ionides and Jesse Wheeler and Kevin Tan},
title = {Pypomp},
url = {https://github.com/pypomp/pypomp},
version = {1.0.1},
year = {2026}
}
Metadata
Release files for pypomp 1.0.1
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| pypomp-1.0.1.tar.gz | 7.4 MB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| pypomp-1.0.1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 13.8 MB
Release files / pypomp-1.0.1.tar.gz
| Download URL | pypomp-1.0.1.tar.gz |
|---|---|
| Size | 7.4 MB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
2ccfcbf9fd1a14ffc8ebdc0cd7a61790ffb3d081a261cd93425cca493ac27314
|
|
BLAKE2b-256 checksum How to use checksums |
13da543718dfd56fb57c17933123fef6723e4eb0cc2fe95beea76bc84ec5a27d
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Sep 7, 2026.
Transparency logRelease files / pypomp-1.0.1-py3-none-any.whl
| Download URL | pypomp-1.0.1-py3-none-any.whl |
|---|---|
| Size | 6.4 MB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
0ff614315facb62b0ecb7efe11d4905267d86bc77cf350da316fcb4ac0460b4d
|
|
BLAKE2b-256 checksum How to use checksums |
8720cb644dbae9c60761710b3885d7e05cb7447623e605b8e772502a8b819102
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Sep 7, 2026.
Transparency log