Module: tfp.substrates.jax.experimental.mcmc

TensorFlow Probability experimental MCMC package.

Functions

default_make_hmc_kernel_fn(...): Generate a hmc without transformation kernel.

gen_make_hmc_kernel_fn(...): Generate a transformed hmc kernel.

gen_make_transform_hmc_kernel_fn(...): Generate a transformed hmc kernel.

make_rwmh_kernel_fn(...): Generate a Random Walk MH kernel.

resample_deterministic_minimum_error(...): Deterministic minimum error resampler for sequential Monte Carlo.

resample_independent(...): Categorical resampler for sequential Monte Carlo.

resample_stratified(...): Stratified resampler for sequential Monte Carlo.

resample_systematic(...): A systematic resampler for sequential Monte Carlo.

sample_sequential_monte_carlo(...): Runs Sequential Monte Carlo to sample from the posterior distribution.

simple_heuristic_tuning(...): Tune the number of steps and scaling of one mutation.