Skip to main content

zeus: Lightning Fast MCMC

Project description

logo

zeus is a Python implementation of the Ensemble Slice Sampling method.

  • Fast & Robust Bayesian Inference,
  • Efficient Markov Chain Monte Carlo (MCMC),
  • Black-box inference, no hand-tuning,
  • Excellent performance in terms of autocorrelation time and convergence rate,
  • Scale to multiple CPUs without any extra effort,
  • Automated Convergence diagnostics.

GitHub arXiv arXiv ascl Build Status License: GPL v3 Documentation Status Downloads

Example

For instance, if you wanted to draw samples from a 10-dimensional Gaussian, you would do something like:

import zeus
import numpy as np

def log_prob(x, ivar):
    return - 0.5 * np.sum(ivar * x**2.0)

nsteps, nwalkers, ndim = 1000, 100, 10
ivar = 1.0 / np.random.rand(ndim)
start = np.random.randn(nwalkers,ndim)

sampler = zeus.EnsembleSampler(nwalkers, ndim, log_prob, args=[ivar])
sampler.run_mcmc(start, nsteps)
chain = sampler.get_chain(flat=True)

Documentation

Read the docs at zeus-mcmc.readthedocs.io

Installation

To install zeus using pip run:

pip install zeus-mcmc

To install zeus in a [Ana]Conda environment use:

conda install -c conda-forge zeus-mcmc

Attribution

Please cite the following papers if you found this code useful in your research:

@article{karamanis2021zeus,
  title={zeus: A Python implementation of Ensemble Slice Sampling for efficient Bayesian parameter inference},
  author={Karamanis, Minas and Beutler, Florian and Peacock, John A},
  journal={arXiv preprint arXiv:2105.03468},
  year={2021}
}

@article{karamanis2020ensemble,
    title = {Ensemble slice sampling: Parallel, black-box and gradient-free inference for correlated & multimodal distributions},
    author = {Karamanis, Minas and Beutler, Florian},
    journal = {arXiv preprint arXiv: 2002.06212},
    year = {2020}
}

Licence

Copyright 2019-2021 Minas Karamanis and contributors.

zeus is free software made available under the GPL-3.0 License. For details see the LICENSE file.

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

zeus-mcmc-2.5.4.tar.gz (35.1 kB view details)

Uploaded Source

Built Distribution

zeus_mcmc-2.5.4-py3-none-any.whl (24.1 kB view details)

Uploaded Python 3

File details

Details for the file zeus-mcmc-2.5.4.tar.gz.

File metadata

  • Download URL: zeus-mcmc-2.5.4.tar.gz
  • Upload date:
  • Size: 35.1 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.9.16

File hashes

Hashes for zeus-mcmc-2.5.4.tar.gz
Algorithm Hash digest
SHA256 594baa90de4ad4488c4db5ed6a0446f7103bc4b3de787f4d7d23c91c9aa88769
MD5 94484a0a92700a27dd1a3c43845dbc69
BLAKE2b-256 15c0248a600ae0f3d1437612821f7c528c895b6a8408a052c352d4cac46dcf94

See more details on using hashes here.

File details

Details for the file zeus_mcmc-2.5.4-py3-none-any.whl.

File metadata

  • Download URL: zeus_mcmc-2.5.4-py3-none-any.whl
  • Upload date:
  • Size: 24.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.9.16

File hashes

Hashes for zeus_mcmc-2.5.4-py3-none-any.whl
Algorithm Hash digest
SHA256 a64a7dae15f413200c6d590a3edfc3b8bc63c6bea3acdf123c0397c8089e123f
MD5 2006a06536e91459bf00df4bbaf2fe84
BLAKE2b-256 48cb99ee8021fa305d65b1fb369fabf96ccc65c60282407390e5355bb9e62f9f

See more details on using hashes here.

Supported by

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