Transformers How to use jerilseb/mnist-classifier with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import MNISTClassifier
model = MNISTClassifier.from_pretrained("jerilseb/mnist-classifier", device_map="auto") 
