jax

2f798902 - Better documentation for jnp.nan_to_num