text: Title: Checkpointing during MCMC
  Body: Hi devs, thanks for your contributions to this tool!
  
  Is it possible to save the MCMC chains while the chains are running? I'm using numpyro on multiple GPUs in an HPC environment and would like to checkpoint my jobs in case of preemption.