Simulation-based inference in JAX
Project description
sbijax
Simulation-based inference in JAX
About
SbiJAX implements several algorithms for simulation-based inference using JAX, Haiku and BlackJAX.
SbiJAX so far implements
- Rejection ABC (
RejectionABC
), - Sequential Monte Carlo ABC (
SMCABC
), - Sequential Neural Likelihood Estimation (
SNL
)
Examples
You can find several self-contained examples on how to use the algorithms in examples
.
Installation
Make sure to have a working JAX
installation. Depending whether you want to use CPU/GPU/TPU,
please follow these instructions.
To install from PyPI, just call the following on the command line:
pip install sbijax
To install the latest GitHub , use:
pip install git+https://github.com/dirmeier/sbijax@<RELEASE>
Author
Simon Dirmeier sfyrbnd @ pm me
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
sbijax-0.0.10.tar.gz
(19.0 kB
view hashes)
Built Distribution
sbijax-0.0.10-py3-none-any.whl
(19.9 kB
view hashes)