NumPyroSampler members# checkpoint_path chunk_suffix chunks_dir clear_resume_state close combine_trees flush list_chunk_paths load_checkpoint load_samples metadata_path pending_write_count run save_checkpoint save_samples_chunk to_datatree uses_directory_stores