@article{250511190, title = {JaxSGMC: Modular stochastic gradient MCMC in JAX}, author = {Stephan Thaler and Paul Fuchs and Ana Cukarska and Julija Zavadlav}, year = {2025}, eprint = {2505.11190}, archivePrefix = {arXiv}, url = {https://arxiv.org/abs/2505.11190v1}, }