using tensorflow_dataset just to import the dataset using numpy for the matrix calculations using matplotlib for literally nothing
This Network should work and start training as soon as you run mnist_nn_fc.py and have every package installed correctly.
This Project can be considered completed
(a book wich might help you: https://www.deeplearningbook.org/ )