Bayeux
Stitching together models and samplers
bayeux lets you write a probabilistic model in JAX and immediately have access to state-of-the-art inference methods. The API aims to be simple, self descriptive, and helpful. Simply provide a log density function (which doesn't even have to be normalized), along with a single point (specified as a pytree) where that log density is finite. Then let bayeux do the rest!
Installation
pip install bayeux-ml
Quickstart
We define a model by providing a log density in JAX. This could be defined using a probabilistic programming language (PPL) like numpyro, PyMC, TFP, distrax, oryx, coix, or directly in JAX.
import bayeux as bx
import jax
normal_density = bx.Model(
log_density=lambda x: -x*x,
test_point=1.)
seed = jax.random.key(0)
opt_results = normal_density.optimize.optax_adam(seed=seed)
# OR!
idata = normal_density.mcmc.numpyro_nuts(seed=seed)
# OR!
surrogate_posterior, loss = normal_density.vi.tfp_factored_surrogate_posterior(seed=seed)
Read more
- Defining models
- Inspecting models
- Testing and debugging
- Also see
bayeuxintegration with numpyro, PyMC, and TFP!
This is not an officially supported Google product.
1 maintainer
Metadata
1 maintainer
Release files for bayeux-ml 0.1.15
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| bayeux_ml-0.1.15.tar.gz | 27.2 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| bayeux_ml-0.1.15-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 69.9 kB
Release files / bayeux_ml-0.1.15.tar.gz
| Download URL | bayeux_ml-0.1.15.tar.gz |
|---|---|
| Size | 27.2 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
4409d036281906384b9d717496d0066989d078dc3616066bf5d9b3cda2308bec
|
|
BLAKE2b-256 checksum How to use checksums |
b22e8645f8529d1b35b17041bba1f97135b0771c52cf8f4db0940a3cc681c56b
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.0.1 CPython/3.11.10
|
Release files / bayeux_ml-0.1.15-py3-none-any.whl
| Download URL | bayeux_ml-0.1.15-py3-none-any.whl |
|---|---|
| Size | 42.6 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
d09fb9ef160a5f0d75239d46adeee9e87c23be1d0b294a48edecbd38ff7d7be1
|
|
BLAKE2b-256 checksum How to use checksums |
3a170c6cbe9e8cca9fcaf5fec560388d4b2d7507f6ba2c304c70c495c28a8542
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.0.1 CPython/3.11.10
|
