How to obtain the inverse mass matrix and the time step size

I am doing warmup in windows of certain intervals and I wish to obtain the time step size and the inverse mass matrix after these warmup windows. How can I do so in Numpyro?

there are several forum threads show you how, e.g. Get number of steps, sizes and acc. prob over all iterations

1 Like