It seems like the NUTS sampler automatically calls .get_sampler_stats, so I think this means it’s impossible to use the text backend with NUTS.
It seems like the NUTS sampler automatically calls .get_sampler_stats, so I think this means it’s impossible to use the text backend with NUTS.