Deep Learning with JAX and Flax/

...

/

Challenge: Loss and Activation function

Challenge: Loss and Activation function

Test your understanding of loss functions.

We'll cover the following...

Challenge 1: Implementation of loss functions

You are required to execute the following built-in functions of JAX on the given logits and labels:

  1. Softmax cross-entropy ...