jax

09edc494 - Add explicit_axes section to the doc