External library integrations for Ray Tune#

PyTorch Lightning (tune.integration.pytorch_lightning)#

TuneReportCheckpointCallback

PyTorch Lightning report and checkpoint callback

XGBoost (tune.integration.xgboost)#

TuneReportCheckpointCallback

XGBoost callback to save checkpoints and report metrics for Ray Tune.

LightGBM (tune.integration.lightgbm)#

TuneReportCheckpointCallback

Creates a callback that reports metrics and checkpoints model.