Tutorial: image classification with JAX and Flax Linen

Learn how to create a simple convolutional network with the Linen API by Flax and train it to recognize handwritten digits.