Como carregar o conjunto de dados MNIST no Python

from keras.datasets import mnist

(train_X, train_y), (test_X, test_y) = mnist.load_data()
Motionless Mandrill