Live training loss plot in Jupyter Notebook for Keras, PyTorch and others - View it on GitHub