Description:

aex 0.0.2

AeX
The following currently works:
import aesara.tensor as at
import aex

srng = at.random.RandomStream(0)

sigma_rv = srng.normal(1.)
mu_rv = srng.normal(0, 1)
Y_rv = srng.normal(mu_rv, sigma_rv)

sampler = aex.prior_sampler(Y_rv, mu_rv)
sampler(rng_key, 1_000_000)

Coming
Sampling from the posterior distribution using Blackjax's NUTS sampler:
sampler = aex.mcmc({Y_rv: 1.}, aex.NUTS())
samples, info = sampler(rng_key, 1000, 1000)

Sampling from the posterior by arbitrarily combining Blackjax step functions:
sampler = aex.mcmc({Y_rv: 1.}, {[mu_rv, sigma_rv]: aex.NUTS(), Y_rv: aex.RMH()})
samples, info = sampler(rng_key, 1000)

Sampling from the posterior predictive distribution:
sampler = aex.posterior_predictive(trace, Y_rv)
sampler(rng_key, 1000)

License

For personal and professional use. You cannot resell or redistribute these repositories in their original state.

Share

• Share on Facebook
• Share on Twitter