Econometrics in jax minimal working examples of standard econometrics tasks implemented with modern optimizers in JAX.