Index

B

badodab() (in module sgmcmcjax.diffusions) baoab() (in module sgmcmcjax.diffusions) build_badodab_kernel() (in module sgmcmcjax.kernels) build_badodabCV_kernel() (in module sgmcmcjax.kernels) build_baoab_kernel() (in module sgmcmcjax.kernels) build_grad_log_post() (in module sgmcmcjax.util) build_gradient_estimation_fn() (in module sgmcmcjax.gradient_estimation) build_gradient_estimation_fn_CV() (in module sgmcmcjax.gradient_estimation) build_gradient_estimation_fn_SVRG() (in module sgmcmcjax.gradient_estimation)

build_psgld_kernel() (in module sgmcmcjax.kernels) build_sghmc_kernel() (in module sgmcmcjax.kernels) build_sghmc_SVRG_kernel() (in module sgmcmcjax.kernels) build_sghmcCV_kernel() (in module sgmcmcjax.kernels) build_sgld_kernel() (in module sgmcmcjax.kernels) build_sgld_SVRG_kernel() (in module sgmcmcjax.kernels) build_sgldAdam_kernel() (in module sgmcmcjax.kernels) build_sgldCV_kernel() (in module sgmcmcjax.kernels) build_sgnht_kernel() (in module sgmcmcjax.kernels) build_sgnhtCV_kernel() (in module sgmcmcjax.kernels)

C

cyclical_schedule() (in module sgmcmcjax.diffusions)

D

diffusion_factory() (in module sgmcmcjax.diffusion_util)

I

imq_KSD() (in module sgmcmcjax.ksd)

K

k_0_fun() (in module sgmcmcjax.ksd)

M

module sgmcmcjax.diffusion_util sgmcmcjax.diffusions sgmcmcjax.gradient_estimation sgmcmcjax.kernels sgmcmcjax.ksd sgmcmcjax.samplers sgmcmcjax.util

P

progress_bar_scan() (in module sgmcmcjax.util)

psgld() (in module sgmcmcjax.diffusions)

R

run_loop() (in module sgmcmcjax.util)

S

sghmc() (in module sgmcmcjax.diffusions) sgld() (in module sgmcmcjax.diffusions) sgldAdam() (in module sgmcmcjax.diffusions) sgmcmc_sampler() (in module sgmcmcjax.samplers) sgmcmcjax.diffusion_util module sgmcmcjax.diffusions module sgmcmcjax.gradient_estimation module

sgmcmcjax.kernels module sgmcmcjax.ksd module sgmcmcjax.samplers module sgmcmcjax.util module sgnht() (in module sgmcmcjax.diffusions)

W

welling_teh_schedule() (in module sgmcmcjax.diffusions)