Types & classes — github.com/alisiahkoohi/Langevin-dynamics

CodeHub MCPcopy Create free account

hub / github.com/alisiahkoohi/Langevin-dynamics / types & classes

Types & classes6 in github.com/alisiahkoohi/Langevin-dynamics

⨍ Functions 26 ◇ Types & classes 6

↓ 2 callersClassLangevinDynamics

LangevinDynamics class for performing Langevin dynamics optimization. Args: x (torch.Tensor): Initial parameter values. func

src/langevin_sampling/samplers.py:6

↓ 2 callersClassMetropolisAdjustedLangevin

A class implementing the Metropolis-Adjusted Langevin algorithm. Args: x (torch.Tensor): Initial input tensor. func (callabl

src/langevin_sampling/samplers.py:112

↓ 2 callersClasspSGLD

Implements pSGLD algorithm based on https://arxiv.org/pdf/1512.07666.pdf Built on the PyTorch RMSprop implementation (https://pytorch.org/doc

src/langevin_sampling/precondSGLD.py:5

↓ 1 callersClassGaussianDistribution

examples/gaussian_LD.py:17

↓ 1 callersClassGaussianDistribution

examples/gaussian_MALA.py:16

Implements SGLD algorithm based on https://www.ics.uci.edu/~welling/publications/papers/stoclangevin_v6.pdf Built on the PyTorch SGD impl

src/langevin_sampling/SGLD.py:5