Better docs for jnp.gradient

Commit

1 year ago

Better docs for jnp.gradient Also remove skip_params option from util.implements, as this was its last usage.