ExamplesΒΆ You can find here all basic detailed examples. Analytical functions Training a network: Flax vs PyTorch Stochastic gradient Langevin dynamics Illustration of several algorithms Thinning MCMC outputs Conformal prediction with Flax and MAPIE Laplace approximations