[tune] Added PyTorch Lightning callbacks to integrations (#10220)

Co-authored-by: Richard Liaw <rliaw@berkeley.edu>

doc/source/tune/api_docs/integration.rst

@@ -0,0 +1,46 @@
+.. _tune-integration:
+
+External library integrations (tune.integration)
+================================================
+
+.. contents::
+    :local:
+    :depth: 1
+
+.. _tune-integration-kubernetes:
+
+Kubernetes (tune.integration.kubernetes)
+----------------------------------------
+
+.. autofunction:: ray.tune.integration.kubernetes.NamespacedKubernetesSyncer
+
+.. _tune-integration-pytorch-lightning:
+
+PyTorch Lightning (tune.integration.pytorch_lightning)
+------------------------------------------------------
+
+.. autoclass:: ray.tune.integration.pytorch_lightning.TuneReportCallback
+
+.. autoclass:: ray.tune.integration.pytorch_lightning.TuneReportCheckpointCallback
+
+.. _tune-integration-torch:
+
+Torch (tune.integration.torch)
+------------------------------
+
+.. autofunction:: ray.tune.integration.torch.DistributedTrainableCreator
+
+.. autofunction:: ray.tune.integration.torch.distributed_checkpoint_dir
+
+.. autofunction:: ray.tune.integration.torch.is_distributed_trainable
+
+.. _tune-integration-wandb:
+
+Weights and Biases (tune.integration.wandb)
+-------------------------------------------
+
+:ref:`See also here <tune-wandb>`.
+
+.. autoclass:: ray.tune.integration.wandb.WandbLogger
+
+.. autofunction:: ray.tune.integration.wandb.wandb_mixin

doc/source/tune/api_docs/overview.rst

@@ -22,6 +22,7 @@ on `Github`_.
    schedulers.rst
    sklearn.rst
    logging.rst
+   integration.rst
    internals.rst
    client.rst
    cli.rst

doc/source/tune/api_docs/trainable.rst

@@ -294,10 +294,13 @@ Ray also offers lightweight integrations to distribute your model training on Ra
 
 
 .. autofunction:: ray.tune.integration.torch.DistributedTrainableCreator
+   :noindex:
 
 .. autofunction:: ray.tune.integration.torch.distributed_checkpoint_dir
+   :noindex:
 
 .. autofunction:: ray.tune.integration.torch.is_distributed_trainable
+   :noindex:
 
 tune.DurableTrainable
 ---------------------