Hello MNIST in 4 frameworks

PyTorch / Lightning

pip install -r requirements/pytorch.txt

# pytorch
python pytorch.py

# lightning
python pytorch_ligtning_mnist.py

pip install -r requirements/tensorflow.txt

# keras
python keras.py

# tensorflow
python tf2.py