jax

6735ea09 - Rename `manual_type=.../.manual_type/.mt` to `manual_axis_type=.../.manual_axis_type/.mat` to avoid confusion with `jnp.array(...).mT`