Can this be used to perform neural network inference, such as utilizing the MNIST dataset? (https://pytorch.org/vision/0.20/generated/torchvision.datasets.MNIST.html)