jax

7e38cbd6 - Better docs for jnp.fromfunction