Basic Modelling Example with MNIST Dataset August 15, 2021 Overview In this blog we will go through how to do modelling in Flax with help of MNIST toy dataset.