mnist_net.py Defines and trains the simple neural network, and extracts and saves its weights as .txt files uart_test_nn.py Randomly picks a test image from the dataset and sends it to the baremetal C application on the PYNQ-Z2 board via UART.