Check out Jax magpit (@13