This makes it easier to start training faster without manually writing your own training loop.