Variational Laplace This is a JAX implementation of A primer on Variational Laplace based on the original MATLAB source.