ML Katas

by: newest upvotes saves

Taking gradients with `grad` II

this year easy (<10 mins) | grad autodiff

By default, `jax.grad` will take the gradient with respect to the first argument of the function. However, in many cases, we will want to take gradients with respect to many of the function's...

1