jax_sgmc jax_sgmc.adaption jax_sgmc.alias jax_sgmc.data jax_sgmc.integrator jax_sgmc.io jax_sgmc.potential jax_sgmc.scheduler jax_sgmc.solver jax_sgmc.util