Skip to content

PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks

License

Notifications You must be signed in to change notification settings

AdamCobb/hamiltorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

72 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

hamiltorch

PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks

  • Perform HMC in user-defined log probabilities and in PyTorch neural networks (objects inheriting from the torch.nn.Module).
  • Available sampling schemes:
    • HMC
    • No-U-Turn Sampler (currently adapts step-size only)
    • Implicit RMHMC
    • Explicit RMHMC
    • Symmetric Split HMC

How to install

pip install git+https://github.com/AdamCobb/hamiltorch

How does it work?

There are currently two blog posts that describe how to use hamiltorch:

There are also notebook-style tutorials:

How to cite?

Please consider citing the following papers if you use hamiltorch in your research:

For symmetric splitting:

@article{cobb2020scaling,
  title={Scaling Hamiltonian Monte Carlo Inference for Bayesian Neural Networks with Symmetric Splitting},
  author={Cobb, Adam D and Jalaian, Brian},
  journal={Uncertainty in Artificial Intelligence},
  year={2021}
}

For RMHMC:

@article{cobb2019introducing,
  title={Introducing an Explicit Symplectic Integration Scheme for Riemannian Manifold Hamiltonian Monte Carlo},
  author={Cobb, Adam D and Baydin, At{\i}l{\i}m G{\"u}ne{\c{s}} and Markham, Andrew and Roberts, Stephen J},
  journal={arXiv preprint arXiv:1910.06243},
  year={2019}
}

Star History

Star History Chart

Who developed hamiltorch?

Adam D Cobb

Atılım Güneş Baydin

Brian Jalaian

About

PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published