Index B | C | D | E | G | H | I | K | L | M | Q | R | S | T | U B batorch.rkhs.discrepancies module batorch.rkhs.kernels module batorch.rkhs.quantization module batorch.sgmcmc.diffusions module batorch.sgmcmc.samplers module batorch.utils.pipelines module C compute_full_center_gradient() (batorch.sgmcmc.samplers.EulerExplicitControlVariates method) compute_gradients_log_target() (batorch.sgmcmc.samplers.Sampler method) D Diagonal() (batorch.rkhs.kernels.EnergyKernel method) (batorch.rkhs.kernels.ImqKernel method) (batorch.rkhs.kernels.Kernel method) Diffusion (class in batorch.sgmcmc.diffusions) DiffusionCyclicSGHMC (class in batorch.sgmcmc.diffusions) DiffusionCyclicSGLD (class in batorch.sgmcmc.diffusions) DiffusionpSGLD (class in batorch.sgmcmc.diffusions) DiffusionSGHMC (class in batorch.sgmcmc.diffusions) DiffusionSGHMCSA (class in batorch.sgmcmc.diffusions) DiffusionSGLD (class in batorch.sgmcmc.diffusions) E EnergyKernel (class in batorch.rkhs.kernels) ensemble_pipeline() (in module hydra_entries) estimate_gradients() (batorch.sgmcmc.samplers.EulerExplicit method) (batorch.sgmcmc.samplers.EulerExplicitControlVariates method) EulerExplicit (class in batorch.sgmcmc.samplers) EulerExplicitControlVariates (class in batorch.sgmcmc.samplers) EulerExplicitFixedPoint (class in batorch.sgmcmc.samplers) G get_grads() (batorch.sgmcmc.samplers.Sampler method) get_params() (batorch.sgmcmc.samplers.Sampler method) GramMatrix() (batorch.rkhs.kernels.EnergyKernel method) (batorch.rkhs.kernels.ImqKernel method) (batorch.rkhs.kernels.Kernel method) H hydra_entries module I ImqKernel (class in batorch.rkhs.kernels) init_state() (batorch.sgmcmc.diffusions.Diffusion method) (batorch.sgmcmc.diffusions.DiffusionCyclicSGHMC method) (batorch.sgmcmc.diffusions.DiffusionCyclicSGLD method) (batorch.sgmcmc.diffusions.DiffusionpSGLD method) (batorch.sgmcmc.diffusions.DiffusionSGHMC method) (batorch.sgmcmc.diffusions.DiffusionSGHMCSA method) (batorch.sgmcmc.diffusions.DiffusionSGLD method) K Kernel (class in batorch.rkhs.kernels) KSD() (in module batorch.rkhs.discrepancies) L Line() (batorch.rkhs.kernels.EnergyKernel method) (batorch.rkhs.kernels.ImqKernel method) (batorch.rkhs.kernels.Kernel method) LineSteinMatrix() (batorch.rkhs.kernels.ImqKernel method) (batorch.rkhs.kernels.Kernel method) M map_pipeline() (in module hydra_entries) mcdropout_pipeline() (in module hydra_entries) mcmc() (in module batorch.utils.pipelines) mcmc_pipeline() (in module hydra_entries) median_heuristic() (in module batorch.rkhs.quantization) MMD() (in module batorch.rkhs.discrepancies) module batorch.rkhs.discrepancies batorch.rkhs.kernels batorch.rkhs.quantization batorch.sgmcmc.diffusions batorch.sgmcmc.samplers batorch.utils.pipelines hydra_entries Q Quantization (class in batorch.rkhs.quantization) QuantizationMMD (class in batorch.rkhs.quantization) R resample_momentum() (batorch.sgmcmc.diffusions.DiffusionCyclicSGHMC method) (batorch.sgmcmc.diffusions.DiffusionSGHMC method) S Sampler (class in batorch.sgmcmc.samplers) SamplerFactory() (in module batorch.sgmcmc.samplers) select() (batorch.rkhs.quantization.Quantization method) (batorch.rkhs.quantization.QuantizationMMD method) set_params() (batorch.sgmcmc.samplers.Sampler method) SlicedKSD() (in module batorch.rkhs.discrepancies) SlicedSteinMatrix() (batorch.rkhs.kernels.ImqKernel method) (batorch.rkhs.kernels.Kernel method) SteinMatrix() (batorch.rkhs.kernels.ImqKernel method) (batorch.rkhs.kernels.Kernel method) step() (batorch.sgmcmc.diffusions.Diffusion method) (batorch.sgmcmc.samplers.EulerExplicit method) (batorch.sgmcmc.samplers.EulerExplicitFixedPoint method) (batorch.sgmcmc.samplers.Sampler method) T test_loop() (in module batorch.utils.pipelines) train_loop() (in module batorch.utils.pipelines) U update_fn() (batorch.sgmcmc.diffusions.Diffusion method) (batorch.sgmcmc.diffusions.DiffusionCyclicSGHMC method) (batorch.sgmcmc.diffusions.DiffusionCyclicSGLD method) (batorch.sgmcmc.diffusions.DiffusionpSGLD method) (batorch.sgmcmc.diffusions.DiffusionSGHMC method) (batorch.sgmcmc.diffusions.DiffusionSGHMCSA method) (batorch.sgmcmc.diffusions.DiffusionSGLD method)