Using text backend raises sampler stats error

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.