SGMCMCJax: a lightweight JAX library for stochastic gradient Markov chain Monte Carlo algorithmsJournal of Open Source Software