diff --git a/docs/cugraph-docs/source/api_docs/cugraph-pyg/cugraph_pyg.rst b/docs/cugraph-docs/source/api_docs/cugraph-pyg/cugraph_pyg.rst
index 3b1cfcd..2552206 100644
--- a/docs/cugraph-docs/source/api_docs/cugraph-pyg/cugraph_pyg.rst
+++ b/docs/cugraph-docs/source/api_docs/cugraph-pyg/cugraph_pyg.rst
@@ -20,6 +20,15 @@ Feature Storage
    
    cugraph_pyg.data.feature_store.FeatureStore
 
+Tensors and Embeddings
+----------------------
+.. autosummary::
+   :toctree ../api/cugraph-pyg/
+
+   cugraph_pyg.tensor.dist_tensor.DistTensor
+   cugraph_pyg.tensor.dist_tensor.DistEmbedding
+   cugraph_pyg.tensor.dist_matrix.DistMatrix
+
 
 Data Loaders
 ------------
@@ -29,6 +38,9 @@ Data Loaders
    cugraph_pyg.loader.node_loader.NodeLoader
    cugraph_pyg.loader.neighbor_loader.NeighborLoader
 
+   cugraph_pyg.loader.link_loader.LinkLoader
+   cugraph_pyg.loader.link_neighbor_loader.LinkNeighborLoader
+
 Samplers
 --------
 .. autosummary::
@@ -37,4 +49,8 @@ Samplers
    cugraph_pyg.sampler.sampler.BaseSampler
    cugraph_pyg.sampler.sampler.SampleReader
    cugraph_pyg.sampler.sampler.HomogeneousSampleReader
+   cugraph_pyg.sampler.sampler.HeterogeneousSampleReader
    cugraph_pyg.sampler.sampler.SampleIterator
+
+   cugraph_pyg.sampler.distributed_sampler.BaseDistributedSampler
+   cugraph_pyg.sampler.distributed_sampler.DistributedNeighborSampler