Add a little interface for executing Python code during sampling

[jobrachem]

This interface can be placed in between two jitted blocks, where the tqdm progress bar is updated at the moment. One usecase would be remote logging to weights and biases during sampling.