All Katas - ML Katas

by: newest upvotes saves

Build a Custom ReLU Activation Function

yesterday easy (<10 mins) | autograd jax activation-function custom-function

Implement the Rectified Linear Unit (ReLU) activation function in JAX. Then, use `jax.grad` to find its derivative. The ReLU function is defined as: $ReLU(x) = max(0, x)$ **Verification:** - For...