Jax implementation for the paper "Sampling-based inference for large linear models, with application to linearised Laplace" - View it on GitHub