norse.task.mnist_pl module