jax

ccabdb29 - Fix typos in docs and an error message