Comparison of the sampling quality using jaxspec and xspec. Left: $\widehat{R}$ statistic computed for 100 runs of 10⁵ steps of NUTS and AIES using the jaxpsec and xspec implementations. The usual convergence threshold of $\widehat{R}<1.01$ is shown in black. Right: value of the parameter norm_BB for 10⁴ steps of a single chain in the sample using NUTS and AIES with jaxpsec and xspec implementations. The solid line represents a 100-step moving average of the parameter value.