Update to use Pytorch with Keras 3 instead of Tensorflow
keras.datasets.mnist.load_data(), Keras team, 2024 - Official Keras documentation on how to load and use the MNIST dataset within a Python environment.
Deep Learning, Ian Goodfellow, Yoshua Bengio, and Aaron Courville, 2016 (MIT Press) - A foundational textbook that covers general machine learning and deep learning concepts, including the characteristics of datasets like MNIST and the importance of data handling and preprocessing for neural networks.