ML Katas

by: newest upvotes saves

Taking gradients with `grad`

this year easy (<30 mins) | grad autodiff

One of JAX's most powerful features is its ability to automatically differentiate your code. In this exercise, you will implement a regression problem and use `jax.grad` to compute gradients. 1....

1