We offer a torch.utils.data.Dataset implementation for power grid data, and some tools and advices on how to accelerate a training loop.