Is it possible to save intermediate warmup samples when using collect_samples=True, similar to how you can save intermediate samples during the normal NUTS sampling using post_warmup_state?
Like the question here I am taking 10,000 samples but saving every 1,000. When I try running warmup for 4,000 warmup samples with collect_warmup=True I have a resource exhausted error (which does not occur when I don’t collect warmup samples).
Thank you