martin-marek / mini-hmc-jax

A simple implementation of Hamiltonian Monte Carlo in JAX.
16Updated 9 months ago

Related projects

Alternatives and complementary repositories for mini-hmc-jax