jax

3a0ee844 - Improve docs for jax.numpy arithmetic comparison operations