/rmhmc

Riemannian HMC in JAX

Primary LanguagePythonMIT LicenseMIT

rmhmc

Riemannian HMC in JAX

Installation

python -m pip install rmhmc