jax documents