rlouf / mcx

Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.
https://rlouf.github.io/mcx
Apache License 2.0
325 stars 17 forks source link

Simplify `HMC` program definition #18

Closed rlouf closed 4 years ago

rlouf commented 4 years ago

To initialize the HMC program we currently need to pass inverse_mass_matrix and mass_matrix_sqrt. This is redundant, and potentially confusing. We should only require mass_matrix.

rlouf commented 4 years ago

Addressed with #27, closing.