ML Katas

by: newest upvotes saves

Working with a NN library: Flax

this year medium (<1 hr) | flax MLP TrainState

While it is possible to build neural networks from scratch in JAX, it is often more convenient to use a library like Flax or Haiku. These libraries provide common neural network layers and...

1