JaxSGMC: Modular stochastic gradient MCMC in JAX

16 May 2025

Papers citing "JaxSGMC: Modular stochastic gradient MCMC in JAX"

2 / 2 papers shown

Title
chemtrain: Learning Deep Potential Models via Automatic Differentiation and Statistical Physics Paul Fuchs Stephan Thaler Sebastien Röcken Julija Zavadlav DiffM 82 6 0 28 Aug 2024
Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learning Y. Gal Zoubin Ghahramani UQCV BDL 285 9,138 0 06 Jun 2015