1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78
|
:github_url: https://github.com/pyg-team/pytorch_geometric
PyG Documentation
=================
:pyg:`null` **PyG** *(PyTorch Geometric)* is a library built upon :pytorch:`null` `PyTorch <https://pytorch.org>`_ to easily write and train Graph Neural Networks (GNNs) for a wide range of applications related to structured data.
It consists of various methods for deep learning on graphs and other irregular structures, also known as `geometric deep learning <http://geometricdeeplearning.com/>`_, from a variety of published papers.
In addition, it consists of easy-to-use mini-batch loaders for operating on many small and single giant graphs, `multi GPU-support <https://github.com/pyg-team/pytorch_geometric/tree/master/examples/multi_gpu>`_, `torch.compile <https://pytorch-geometric.readthedocs.io/en/latest/advanced/compile.html>`_ support, `DataPipe <https://github.com/pyg-team/pytorch_geometric/blob/master/examples/datapipe.py>`_ support, a large number of common benchmark datasets (based on simple interfaces to create your own), and helpful transforms, both for learning on arbitrary graphs as well as on 3D meshes or point clouds.
.. slack_button::
.. toctree::
:maxdepth: 1
:caption: Install PyG
install/installation
.. toctree::
:maxdepth: 1
:caption: Get Started
get_started/introduction
get_started/colabs
.. toctree::
:maxdepth: 1
:caption: Tutorials
tutorial/gnn_design
tutorial/dataset
tutorial/application
tutorial/distributed
.. toctree::
:maxdepth: 1
:caption: Advanced Concepts
advanced/batching
advanced/sparse_tensor
advanced/hgam
advanced/compile
advanced/jit
advanced/remote
advanced/graphgym
advanced/cpu_affinity
.. toctree::
:maxdepth: 1
:caption: Package Reference
modules/root
modules/nn
modules/data
modules/loader
modules/sampler
modules/datasets
modules/transforms
modules/utils
modules/explain
modules/metrics
modules/distributed
modules/contrib
modules/graphgym
modules/profile
.. toctree::
:maxdepth: 1
:caption: Cheatsheets
cheatsheet/gnn_cheatsheet
cheatsheet/data_cheatsheet
.. toctree::
:maxdepth: 1
:caption: External Resources
external/resources
|