Jax @ MAXIM Models