JaxSGMC provides modular, jit-compilable stochastic gradient MCMC samplers for JAX, including pSGLD, SGHMC, replica exchange, AMAGOLD, and SGGMC.
Title resolution pending
1 Pith paper cite this work. Polarity classification is still indexing.
1
Pith paper citing it
fields
stat.CO 1years
2025 1verdicts
CONDITIONAL 1representative citing papers
citing papers explorer
-
JaxSGMC: Modular stochastic gradient MCMC in JAX
JaxSGMC provides modular, jit-compilable stochastic gradient MCMC samplers for JAX, including pSGLD, SGHMC, replica exchange, AMAGOLD, and SGGMC.