Documentation contents:
API documentation
StragglerDetectionCallback
StragglerDetectionCallback.on_train_batch_end()
StragglerDetectionCallback.setup()
StragglerDetectionCallback.teardown()