lightning/examples/pl_ipu
..
mnist_sample.py