jax

f2c2892c - Refactor jax.numpy docstring tests