Scripts for the live demo part of the PyTorch on LUMI talk at e-INFRA CZ conference, April 2024, Prague - View it on GitHub