gitextract_yj2ugid6/ ├── .github/ │ ├── CODEOWNERS │ ├── CONTRIBUTING.md │ ├── ISSUE_TEMPLATE/ │ │ ├── bug-report.yml │ │ ├── config.yml │ │ ├── documentation.yml │ │ ├── feature-request.yml │ │ ├── installation.yml │ │ └── refactor.yml │ ├── actions/ │ │ └── setup/ │ │ └── action.yml │ ├── dependabot.yml │ ├── labeler.yml │ └── workflows/ │ ├── _testing.yml │ ├── auto-merge.yml │ ├── building_nightly.yml │ ├── changelog.yml │ ├── documentation.yml │ ├── examples.yml │ ├── labeler.yml │ ├── linting.yml │ ├── testing.yml │ └── testing_rag.yml ├── .gitignore ├── .pre-commit-config.yaml ├── CHANGELOG.md ├── CITATION.cff ├── LICENSE ├── README.md ├── benchmark/ │ ├── README.md │ ├── citation/ │ │ ├── README.md │ │ ├── __init__.py │ │ ├── appnp.py │ │ ├── arma.py │ │ ├── cheb.py │ │ ├── datasets.py │ │ ├── gat.py │ │ ├── gcn.py │ │ ├── inference.sh │ │ ├── run.sh │ │ ├── sgc.py │ │ ├── statistics.py │ │ └── train_eval.py │ ├── inference/ │ │ ├── README.md │ │ └── inference_benchmark.py │ ├── kernel/ │ │ ├── README.md │ │ ├── __init__.py │ │ ├── asap.py │ │ ├── datasets.py │ │ ├── diff_pool.py │ │ ├── edge_pool.py │ │ ├── gcn.py │ │ ├── gin.py │ │ ├── global_attention.py │ │ ├── graclus.py │ │ ├── graph_sage.py │ │ ├── main.py │ │ ├── main_performance.py │ │ ├── sag_pool.py │ │ ├── set2set.py │ │ ├── sort_pool.py │ │ ├── statistics.py │ │ ├── top_k.py │ │ └── train_eval.py │ ├── loader/ │ │ └── neighbor_loader.py │ ├── multi_gpu/ │ │ └── training/ │ │ ├── README.md │ │ ├── common.py │ │ ├── training_benchmark_cuda.py │ │ └── training_benchmark_xpu.py │ ├── points/ │ │ ├── README.md │ │ ├── __init__.py │ │ ├── datasets.py │ │ ├── edge_cnn.py │ │ ├── mpnn.py │ │ ├── point_cnn.py │ │ ├── point_net.py │ │ ├── spline_cnn.py │ │ ├── statistics.py │ │ └── train_eval.py │ ├── runtime/ │ │ ├── README.md │ │ ├── __init__.py │ │ ├── dgl/ │ │ │ ├── gat.py │ │ │ ├── gcn.py │ │ │ ├── hidden.py │ │ │ ├── main.py │ │ │ ├── rgcn.py │ │ │ └── train.py │ │ ├── gat.py │ │ ├── gcn.py │ │ ├── main.py │ │ ├── rgcn.py │ │ └── train.py │ ├── setup.py │ ├── training/ │ │ ├── README.md │ │ └── training_benchmark.py │ └── utils/ │ ├── __init__.py │ ├── hetero_gat.py │ ├── hetero_sage.py │ └── utils.py ├── codecov.yml ├── docker/ │ ├── Dockerfile │ ├── Dockerfile.xpu │ ├── README.md │ └── singularity ├── docs/ │ ├── Makefile │ ├── README.md │ ├── requirements.txt │ └── source/ │ ├── .gitignore │ ├── _figures/ │ │ ├── .gitignore │ │ ├── build.sh │ │ ├── graph.tex │ │ ├── hg_example.tex │ │ ├── to_hetero.tex │ │ └── to_hetero_with_bases.tex │ ├── _static/ │ │ └── js/ │ │ └── version_alert.js │ ├── _templates/ │ │ └── autosummary/ │ │ ├── class.rst │ │ ├── inherited_class.rst │ │ ├── metrics.rst │ │ ├── nn.rst │ │ └── only_class.rst │ ├── advanced/ │ │ ├── batching.rst │ │ ├── compile.rst │ │ ├── cpu_affinity.rst │ │ ├── graphgym.rst │ │ ├── hgam.rst │ │ ├── jit.rst │ │ ├── remote.rst │ │ └── sparse_tensor.rst │ ├── cheatsheet/ │ │ ├── data_cheatsheet.rst │ │ └── gnn_cheatsheet.rst │ ├── conf.py │ ├── external/ │ │ └── resources.rst │ ├── get_started/ │ │ ├── colabs.rst │ │ └── introduction.rst │ ├── index.rst │ ├── install/ │ │ ├── installation.rst │ │ └── quick-start.html │ ├── modules/ │ │ ├── contrib.rst │ │ ├── data.rst │ │ ├── datasets.rst │ │ ├── distributed.rst │ │ ├── explain.rst │ │ ├── graphgym.rst │ │ ├── llm.rst │ │ ├── loader.rst │ │ ├── metrics.rst │ │ ├── nn.rst │ │ ├── profile.rst │ │ ├── root.rst │ │ ├── sampler.rst │ │ ├── transforms.rst │ │ └── utils.rst │ ├── notes/ │ │ ├── batching.rst │ │ ├── cheatsheet.rst │ │ ├── colabs.rst │ │ ├── create_dataset.rst │ │ ├── create_gnn.rst │ │ ├── data_cheatsheet.rst │ │ ├── explain.rst │ │ ├── graphgym.rst │ │ ├── heterogeneous.rst │ │ ├── installation.rst │ │ ├── introduction.rst │ │ ├── jit.rst │ │ ├── load_csv.rst │ │ ├── remote.rst │ │ ├── resources.rst │ │ └── sparse_tensor.rst │ └── tutorial/ │ ├── application.rst │ ├── compile.rst │ ├── create_dataset.rst │ ├── create_gnn.rst │ ├── dataset.rst │ ├── dataset_splitting.rst │ ├── distributed.rst │ ├── distributed_pyg.rst │ ├── explain.rst │ ├── gnn_design.rst │ ├── graph_transformer.rst │ ├── heterogeneous.rst │ ├── load_csv.rst │ ├── multi_gpu_vanilla.rst │ ├── multi_node_multi_gpu_vanilla.rst │ ├── neighbor_loader.rst │ ├── point_cloud.rst │ └── shallow_node_embeddings.rst ├── examples/ │ ├── README.md │ ├── agnn.py │ ├── ar_link_pred.py │ ├── argva_node_clustering.py │ ├── arma.py │ ├── attentive_fp.py │ ├── autoencoder.py │ ├── cluster_gcn_ppi.py │ ├── cluster_gcn_reddit.py │ ├── colors_topk_pool.py │ ├── compile/ │ │ ├── gcn.py │ │ └── gin.py │ ├── contrib/ │ │ ├── README.md │ │ ├── pgm_explainer_graph_classification.py │ │ ├── pgm_explainer_node_classification.py │ │ ├── rbcd_attack.py │ │ └── rbcd_attack_poisoning.py │ ├── cora.py │ ├── correct_and_smooth.py │ ├── cpp/ │ │ ├── CMakeLists.txt │ │ ├── README.md │ │ ├── main.cpp │ │ └── save_model.py │ ├── datapipe.py │ ├── dgcnn_classification.py │ ├── dgcnn_segmentation.py │ ├── dir_gnn.py │ ├── distributed/ │ │ ├── README.md │ │ ├── graphlearn_for_pytorch/ │ │ │ ├── README.md │ │ │ ├── dist_train_sage_sup_config.yml │ │ │ ├── dist_train_sage_supervised.py │ │ │ ├── launch.py │ │ │ └── partition_ogbn_dataset.py │ │ ├── kuzu/ │ │ │ ├── README.md │ │ │ └── papers_100M/ │ │ │ ├── README.md │ │ │ ├── prepare_data.py │ │ │ └── train.py │ │ └── pyg/ │ │ └── README.md │ ├── dna.py │ ├── egc.py │ ├── equilibrium_median.py │ ├── explain/ │ │ ├── README.md │ │ ├── captum_explainer.py │ │ ├── captum_explainer_hetero_link.py │ │ ├── gnn_explainer.py │ │ ├── gnn_explainer_ba_shapes.py │ │ ├── gnn_explainer_link_pred.py │ │ └── graphmask_explainer.py │ ├── faust.py │ ├── film.py │ ├── gat.py │ ├── gcn.py │ ├── gcn2_cora.py │ ├── gcn2_ppi.py │ ├── geniepath.py │ ├── glnn.py │ ├── gpse.py │ ├── graph_gps.py │ ├── graph_sage_unsup.py │ ├── graph_sage_unsup_ppi.py │ ├── graph_saint.py │ ├── graph_unet.py │ ├── hetero/ │ │ ├── README.md │ │ ├── bipartite_sage.py │ │ ├── bipartite_sage_unsup.py │ │ ├── dmgi_unsup.py │ │ ├── han_imdb.py │ │ ├── hetero_conv_dblp.py │ │ ├── hetero_link_pred.py │ │ ├── hgt_dblp.py │ │ ├── hierarchical_sage.py │ │ ├── load_csv.py │ │ ├── metapath2vec.py │ │ ├── recommender_system.py │ │ ├── temporal_link_pred.py │ │ └── to_hetero_mag.py │ ├── hierarchical_sampling.py │ ├── infomax_inductive.py │ ├── infomax_transductive.py │ ├── jit/ │ │ ├── README.md │ │ ├── film.py │ │ ├── gat.py │ │ ├── gcn.py │ │ └── gin.py │ ├── kge_fb15k_237.py │ ├── label_prop.py │ ├── lcm_aggr_2nd_min.py │ ├── lightgcn.py │ ├── link_pred.py │ ├── linkx.py │ ├── llm/ │ │ ├── README.md │ │ ├── g_retriever.py │ │ ├── git_mol.py │ │ ├── glem.py │ │ ├── molecule_gpt.py │ │ ├── protein_mpnn.py │ │ └── txt2kg_rag.py │ ├── lpformer.py │ ├── mem_pool.py │ ├── mixhop.py │ ├── mnist_graclus.py │ ├── mnist_nn_conv.py │ ├── mnist_voxel_grid.py │ ├── multi_gpu/ │ │ ├── README.md │ │ ├── distributed_batching.py │ │ ├── distributed_sampling.py │ │ ├── distributed_sampling_multinode.py │ │ ├── distributed_sampling_multinode.sbatch │ │ ├── distributed_sampling_xpu.py │ │ ├── mag240m_graphsage.py │ │ ├── model_parallel.py │ │ ├── papers100m_gcn.py │ │ ├── papers100m_gcn_multinode.py │ │ ├── pcqm4m_ogb.py │ │ └── taobao.py │ ├── mutag_gin.py │ ├── node2vec.py │ ├── ogbn_proteins_deepgcn.py │ ├── ogbn_train.py │ ├── ogc.py │ ├── pmlp.py │ ├── pna.py │ ├── point_transformer_classification.py │ ├── point_transformer_segmentation.py │ ├── pointnet2_classification.py │ ├── pointnet2_segmentation.py │ ├── ppi.py │ ├── proteins_diff_pool.py │ ├── proteins_dmon_pool.py │ ├── proteins_gmt.py │ ├── proteins_mincut_pool.py │ ├── proteins_topk_pool.py │ ├── pytorch_ignite/ │ │ ├── README.md │ │ └── gin.py │ ├── pytorch_lightning/ │ │ ├── README.md │ │ ├── gin.py │ │ ├── graph_sage.py │ │ └── relational_gnn.py │ ├── qm9_nn_conv.py │ ├── qm9_pretrained_dimenet.py │ ├── qm9_pretrained_schnet.py │ ├── quiver/ │ │ ├── README.md │ │ ├── multi_gpu_quiver.py │ │ └── single_gpu_quiver.py │ ├── randlanet_classification.py │ ├── randlanet_segmentation.py │ ├── rdl.py │ ├── rect.py │ ├── reddit.py │ ├── renet.py │ ├── rev_gnn.py │ ├── rgat.py │ ├── rgcn.py │ ├── rgcn_link_pred.py │ ├── seal_link_pred.py │ ├── sgc.py │ ├── shadow.py │ ├── sign.py │ ├── signed_gcn.py │ ├── super_gat.py │ ├── tagcn.py │ ├── tensorboard_logging.py │ ├── tgn.py │ ├── triangles_sag_pool.py │ ├── unimp_arxiv.py │ ├── upfd.py │ └── wl_kernel.py ├── graphgym/ │ ├── agg_batch.py │ ├── configs/ │ │ ├── example.yaml │ │ └── pyg/ │ │ ├── example_graph.yaml │ │ ├── example_link.yaml │ │ └── example_node.yaml │ ├── configs_gen.py │ ├── custom_graphgym/ │ │ ├── __init__.py │ │ ├── act/ │ │ │ ├── __init__.py │ │ │ └── example.py │ │ ├── config/ │ │ │ ├── __init__.py │ │ │ └── example.py │ │ ├── encoder/ │ │ │ ├── __init__.py │ │ │ └── example.py │ │ ├── head/ │ │ │ ├── __init__.py │ │ │ └── example.py │ │ ├── layer/ │ │ │ ├── __init__.py │ │ │ └── example.py │ │ ├── loader/ │ │ │ ├── __init__.py │ │ │ └── example.py │ │ ├── loss/ │ │ │ ├── __init__.py │ │ │ └── example.py │ │ ├── network/ │ │ │ ├── __init__.py │ │ │ └── example.py │ │ ├── optimizer/ │ │ │ ├── __init__.py │ │ │ └── example.py │ │ ├── pooling/ │ │ │ ├── __init__.py │ │ │ └── example.py │ │ ├── stage/ │ │ │ ├── __init__.py │ │ │ └── example.py │ │ ├── train/ │ │ │ ├── __init__.py │ │ │ └── example.py │ │ └── transform/ │ │ └── __init__.py │ ├── grids/ │ │ ├── example.txt │ │ └── pyg/ │ │ └── example.txt │ ├── main.py │ ├── parallel.sh │ ├── run_batch.sh │ ├── run_single.sh │ └── sample/ │ ├── dimensions.txt │ └── dimensionsatt.txt ├── pyproject.toml ├── readthedocs.yml ├── test/ │ ├── conftest.py │ ├── contrib/ │ │ ├── explain/ │ │ │ └── test_pgm_explainer.py │ │ └── nn/ │ │ └── models/ │ │ └── test_rbcd_attack.py │ ├── data/ │ │ ├── lightning/ │ │ │ └── test_datamodule.py │ │ ├── test_batch.py │ │ ├── test_data.py │ │ ├── test_database.py │ │ ├── test_datapipes.py │ │ ├── test_dataset.py │ │ ├── test_dataset_summary.py │ │ ├── test_feature_store.py │ │ ├── test_graph_store.py │ │ ├── test_hetero_data.py │ │ ├── test_hypergraph_data.py │ │ ├── test_inherit.py │ │ ├── test_on_disk_dataset.py │ │ ├── test_remote_backend_utils.py │ │ ├── test_storage.py │ │ ├── test_temporal.py │ │ └── test_view.py │ ├── datasets/ │ │ ├── graph_generator/ │ │ │ ├── test_ba_graph.py │ │ │ ├── test_er_graph.py │ │ │ ├── test_grid_graph.py │ │ │ └── test_tree_graph.py │ │ ├── motif_generator/ │ │ │ ├── test_custom_motif.py │ │ │ ├── test_cycle_motif.py │ │ │ ├── test_grid_motif.py │ │ │ └── test_house_motif.py │ │ ├── test_ba_shapes.py │ │ ├── test_bzr.py │ │ ├── test_elliptic.py │ │ ├── test_enzymes.py │ │ ├── test_explainer_dataset.py │ │ ├── test_fake.py │ │ ├── test_git_mol_dataset.py │ │ ├── test_imdb_binary.py │ │ ├── test_infection_dataset.py │ │ ├── test_karate.py │ │ ├── test_medshapenet.py │ │ ├── test_molecule_gpt_dataset.py │ │ ├── test_mutag.py │ │ ├── test_planetoid.py │ │ ├── test_protein_mpnn_dataset.py │ │ ├── test_snap_dataset.py │ │ ├── test_suite_sparse.py │ │ ├── test_tag_dataset.py │ │ ├── test_teeth3ds.py │ │ └── test_web_qsp_dataset.py │ ├── distributed/ │ │ ├── test_dist_link_neighbor_loader.py │ │ ├── test_dist_link_neighbor_sampler.py │ │ ├── test_dist_neighbor_loader.py │ │ ├── test_dist_neighbor_sampler.py │ │ ├── test_dist_utils.py │ │ ├── test_local_feature_store.py │ │ ├── test_local_graph_store.py │ │ ├── test_partition.py │ │ └── test_rpc.py │ ├── explain/ │ │ ├── algorithm/ │ │ │ ├── test_attention_explainer.py │ │ │ ├── test_captum.py │ │ │ ├── test_captum_explainer.py │ │ │ ├── test_captum_hetero.py │ │ │ ├── test_explain_algorithm_utils.py │ │ │ ├── test_gnn_explainer.py │ │ │ ├── test_graphmask_explainer.py │ │ │ └── test_pg_explainer.py │ │ ├── conftest.py │ │ ├── metric/ │ │ │ ├── test_basic_metric.py │ │ │ ├── test_faithfulness.py │ │ │ └── test_fidelity.py │ │ ├── test_explain_config.py │ │ ├── test_explainer.py │ │ ├── test_explanation.py │ │ ├── test_hetero_explainer.py │ │ └── test_hetero_explanation.py │ ├── graphgym/ │ │ ├── example_node.yml │ │ ├── test_config.py │ │ ├── test_graphgym.py │ │ ├── test_logger.py │ │ └── test_register.py │ ├── io/ │ │ ├── example1.off │ │ ├── example2.off │ │ ├── test_fs.py │ │ └── test_off.py │ ├── llm/ │ │ ├── conftest.py │ │ ├── models/ │ │ │ ├── test_g_retriever.py │ │ │ ├── test_git_mol.py │ │ │ ├── test_glem.py │ │ │ ├── test_llm.py │ │ │ ├── test_llm_judge.py │ │ │ ├── test_molecule_gpt.py │ │ │ ├── test_protein_mpnn.py │ │ │ ├── test_sentence_transformer.py │ │ │ ├── test_txt2kg.py │ │ │ └── test_vision_transformer.py │ │ ├── test_large_graph_indexer.py │ │ ├── test_rag_loader.py │ │ └── utils/ │ │ ├── test_rag_backend_utils.py │ │ ├── test_rag_feature_store.py │ │ ├── test_rag_graph_store.py │ │ └── test_vectorrag.py │ ├── loader/ │ │ ├── test_cache.py │ │ ├── test_cluster.py │ │ ├── test_dataloader.py │ │ ├── test_dynamic_batch_sampler.py │ │ ├── test_graph_saint.py │ │ ├── test_hgt_loader.py │ │ ├── test_ibmb_loader.py │ │ ├── test_imbalanced_sampler.py │ │ ├── test_link_neighbor_loader.py │ │ ├── test_mixin.py │ │ ├── test_neighbor_loader.py │ │ ├── test_neighbor_sampler.py │ │ ├── test_prefetch.py │ │ ├── test_random_node_loader.py │ │ ├── test_shadow.py │ │ ├── test_temporal_dataloader.py │ │ ├── test_utils.py │ │ └── test_zip_loader.py │ ├── metrics/ │ │ └── test_link_pred_metric.py │ ├── my_config.yaml │ ├── nn/ │ │ ├── aggr/ │ │ │ ├── test_aggr_utils.py │ │ │ ├── test_attention.py │ │ │ ├── test_basic.py │ │ │ ├── test_deep_sets.py │ │ │ ├── test_equilibrium.py │ │ │ ├── test_fused.py │ │ │ ├── test_gmt.py │ │ │ ├── test_gru.py │ │ │ ├── test_lcm.py │ │ │ ├── test_lstm.py │ │ │ ├── test_mlp_aggr.py │ │ │ ├── test_multi.py │ │ │ ├── test_patch_transformer.py │ │ │ ├── test_quantile.py │ │ │ ├── test_scaler.py │ │ │ ├── test_set2set.py │ │ │ ├── test_set_transformer.py │ │ │ ├── test_sort.py │ │ │ └── test_variance_preserving.py │ │ ├── attention/ │ │ │ ├── test_performer_attention.py │ │ │ ├── test_polynormer_attention.py │ │ │ └── test_qformer.py │ │ ├── conv/ │ │ │ ├── cugraph/ │ │ │ │ ├── test_cugraph_gat_conv.py │ │ │ │ ├── test_cugraph_rgcn_conv.py │ │ │ │ └── test_cugraph_sage_conv.py │ │ │ ├── test_agnn_conv.py │ │ │ ├── test_antisymmetric_conv.py │ │ │ ├── test_appnp.py │ │ │ ├── test_arma_conv.py │ │ │ ├── test_cg_conv.py │ │ │ ├── test_cheb_conv.py │ │ │ ├── test_cluster_gcn_conv.py │ │ │ ├── test_create_gnn.py │ │ │ ├── test_dir_gnn_conv.py │ │ │ ├── test_dna_conv.py │ │ │ ├── test_edge_conv.py │ │ │ ├── test_eg_conv.py │ │ │ ├── test_fa_conv.py │ │ │ ├── test_feast_conv.py │ │ │ ├── test_film_conv.py │ │ │ ├── test_fused_gat_conv.py │ │ │ ├── test_gat_conv.py │ │ │ ├── test_gated_graph_conv.py │ │ │ ├── test_gatv2_conv.py │ │ │ ├── test_gcn2_conv.py │ │ │ ├── test_gcn_conv.py │ │ │ ├── test_gen_conv.py │ │ │ ├── test_general_conv.py │ │ │ ├── test_gin_conv.py │ │ │ ├── test_gmm_conv.py │ │ │ ├── test_gps_conv.py │ │ │ ├── test_graph_conv.py │ │ │ ├── test_gravnet_conv.py │ │ │ ├── test_han_conv.py │ │ │ ├── test_heat_conv.py │ │ │ ├── test_hetero_conv.py │ │ │ ├── test_hgt_conv.py │ │ │ ├── test_hypergraph_conv.py │ │ │ ├── test_le_conv.py │ │ │ ├── test_lg_conv.py │ │ │ ├── test_meshcnn_conv.py │ │ │ ├── test_message_passing.py │ │ │ ├── test_mf_conv.py │ │ │ ├── test_mixhop_conv.py │ │ │ ├── test_nn_conv.py │ │ │ ├── test_pan_conv.py │ │ │ ├── test_pdn_conv.py │ │ │ ├── test_pna_conv.py │ │ │ ├── test_point_conv.py │ │ │ ├── test_point_gnn_conv.py │ │ │ ├── test_point_transformer_conv.py │ │ │ ├── test_ppf_conv.py │ │ │ ├── test_res_gated_graph_conv.py │ │ │ ├── test_rgat_conv.py │ │ │ ├── test_rgcn_conv.py │ │ │ ├── test_sage_conv.py │ │ │ ├── test_sg_conv.py │ │ │ ├── test_signed_conv.py │ │ │ ├── test_simple_conv.py │ │ │ ├── test_spline_conv.py │ │ │ ├── test_ssg_conv.py │ │ │ ├── test_static_graph.py │ │ │ ├── test_supergat_conv.py │ │ │ ├── test_tag_conv.py │ │ │ ├── test_transformer_conv.py │ │ │ ├── test_wl_conv.py │ │ │ ├── test_wl_conv_continuous.py │ │ │ ├── test_x_conv.py │ │ │ └── utils/ │ │ │ └── test_gnn_cheatsheet.py │ │ ├── dense/ │ │ │ ├── test_dense_gat_conv.py │ │ │ ├── test_dense_gcn_conv.py │ │ │ ├── test_dense_gin_conv.py │ │ │ ├── test_dense_graph_conv.py │ │ │ ├── test_dense_sage_conv.py │ │ │ ├── test_diff_pool.py │ │ │ ├── test_dmon_pool.py │ │ │ ├── test_linear.py │ │ │ └── test_mincut_pool.py │ │ ├── functional/ │ │ │ ├── test_bro.py │ │ │ └── test_gini.py │ │ ├── kge/ │ │ │ ├── test_complex.py │ │ │ ├── test_distmult.py │ │ │ ├── test_rotate.py │ │ │ └── test_transe.py │ │ ├── models/ │ │ │ ├── test_attentive_fp.py │ │ │ ├── test_attract_repel.py │ │ │ ├── test_autoencoder.py │ │ │ ├── test_basic_gnn.py │ │ │ ├── test_correct_and_smooth.py │ │ │ ├── test_deep_graph_infomax.py │ │ │ ├── test_deepgcn.py │ │ │ ├── test_dimenet.py │ │ │ ├── test_gnnff.py │ │ │ ├── test_gpse.py │ │ │ ├── test_graph_mixer.py │ │ │ ├── test_graph_unet.py │ │ │ ├── test_jumping_knowledge.py │ │ │ ├── test_label_prop.py │ │ │ ├── test_lightgcn.py │ │ │ ├── test_linkx.py │ │ │ ├── test_lpformer.py │ │ │ ├── test_mask_label.py │ │ │ ├── test_meta.py │ │ │ ├── test_metapath2vec.py │ │ │ ├── test_mlp.py │ │ │ ├── test_neural_fingerprint.py │ │ │ ├── test_node2vec.py │ │ │ ├── test_pmlp.py │ │ │ ├── test_polynormer.py │ │ │ ├── test_re_net.py │ │ │ ├── test_rect.py │ │ │ ├── test_rev_gnn.py │ │ │ ├── test_schnet.py │ │ │ ├── test_sgformer.py │ │ │ ├── test_signed_gcn.py │ │ │ ├── test_tgn.py │ │ │ └── test_visnet.py │ │ ├── norm/ │ │ │ ├── test_batch_norm.py │ │ │ ├── test_diff_group_norm.py │ │ │ ├── test_graph_norm.py │ │ │ ├── test_graph_size_norm.py │ │ │ ├── test_instance_norm.py │ │ │ ├── test_layer_norm.py │ │ │ ├── test_mean_subtraction_norm.py │ │ │ ├── test_msg_norm.py │ │ │ └── test_pair_norm.py │ │ ├── pool/ │ │ │ ├── connect/ │ │ │ │ └── test_filter_edges.py │ │ │ ├── select/ │ │ │ │ └── test_select_topk.py │ │ │ ├── test_approx_knn.py │ │ │ ├── test_asap.py │ │ │ ├── test_avg_pool.py │ │ │ ├── test_cluster_pool.py │ │ │ ├── test_consecutive.py │ │ │ ├── test_decimation.py │ │ │ ├── test_edge_pool.py │ │ │ ├── test_glob.py │ │ │ ├── test_graclus.py │ │ │ ├── test_knn.py │ │ │ ├── test_max_pool.py │ │ │ ├── test_mem_pool.py │ │ │ ├── test_pan_pool.py │ │ │ ├── test_pool.py │ │ │ ├── test_sag_pool.py │ │ │ ├── test_topk_pool.py │ │ │ └── test_voxel_grid.py │ │ ├── test_compile_basic.py │ │ ├── test_compile_conv.py │ │ ├── test_compile_dynamic.py │ │ ├── test_data_parallel.py │ │ ├── test_encoding.py │ │ ├── test_fvcore.py │ │ ├── test_fx.py │ │ ├── test_inits.py │ │ ├── test_model_hub.py │ │ ├── test_model_summary.py │ │ ├── test_module_dict.py │ │ ├── test_parameter_dict.py │ │ ├── test_reshape.py │ │ ├── test_resolver.py │ │ ├── test_sequential.py │ │ ├── test_to_fixed_size_transformer.py │ │ ├── test_to_hetero_module.py │ │ ├── test_to_hetero_transformer.py │ │ ├── test_to_hetero_with_bases_transformer.py │ │ └── unpool/ │ │ └── test_knn_interpolate.py │ ├── profile/ │ │ ├── test_benchmark.py │ │ ├── test_nvtx.py │ │ ├── test_profile.py │ │ ├── test_profile_utils.py │ │ └── test_profiler.py │ ├── sampler/ │ │ ├── test_sampler_base.py │ │ └── test_sampler_neighbor_sampler.py │ ├── test_config_mixin.py │ ├── test_config_store.py │ ├── test_debug.py │ ├── test_edge_index.py │ ├── test_experimental.py │ ├── test_hash_tensor.py │ ├── test_home.py │ ├── test_index.py │ ├── test_inspector.py │ ├── test_isinstance.py │ ├── test_onnx.py │ ├── test_seed.py │ ├── test_typing.py │ ├── test_warnings.py │ ├── testing/ │ │ └── test_decorators.py │ ├── transforms/ │ │ ├── test_add_gpse.py │ │ ├── test_add_metapaths.py │ │ ├── test_add_positional_encoding.py │ │ ├── test_add_remaining_self_loops.py │ │ ├── test_add_self_loops.py │ │ ├── test_cartesian.py │ │ ├── test_center.py │ │ ├── test_compose.py │ │ ├── test_constant.py │ │ ├── test_delaunay.py │ │ ├── test_distance.py │ │ ├── test_face_to_edge.py │ │ ├── test_feature_propagation.py │ │ ├── test_fixed_points.py │ │ ├── test_gcn_norm.py │ │ ├── test_gdc.py │ │ ├── test_generate_mesh_normals.py │ │ ├── test_grid_sampling.py │ │ ├── test_half_hop.py │ │ ├── test_knn_graph.py │ │ ├── test_laplacian_lambda_max.py │ │ ├── test_largest_connected_components.py │ │ ├── test_line_graph.py │ │ ├── test_linear_transformation.py │ │ ├── test_local_cartesian.py │ │ ├── test_local_degree_profile.py │ │ ├── test_mask_transform.py │ │ ├── test_node_property_split.py │ │ ├── test_normalize_features.py │ │ ├── test_normalize_rotation.py │ │ ├── test_normalize_scale.py │ │ ├── test_one_hot_degree.py │ │ ├── test_pad.py │ │ ├── test_point_pair_features.py │ │ ├── test_polar.py │ │ ├── test_radius_graph.py │ │ ├── test_random_flip.py │ │ ├── test_random_jitter.py │ │ ├── test_random_link_split.py │ │ ├── test_random_node_split.py │ │ ├── test_random_rotate.py │ │ ├── test_random_scale.py │ │ ├── test_random_shear.py │ │ ├── test_remove_duplicated_edges.py │ │ ├── test_remove_isolated_nodes.py │ │ ├── test_remove_self_loops.py │ │ ├── test_remove_training_classes.py │ │ ├── test_rooted_subgraph.py │ │ ├── test_sample_points.py │ │ ├── test_sign.py │ │ ├── test_spherical.py │ │ ├── test_svd_feature_reduction.py │ │ ├── test_target_indegree.py │ │ ├── test_to_dense.py │ │ ├── test_to_device.py │ │ ├── test_to_sparse_tensor.py │ │ ├── test_to_superpixels.py │ │ ├── test_to_undirected.py │ │ ├── test_two_hop.py │ │ └── test_virtual_node.py │ ├── utils/ │ │ ├── conftest.py │ │ ├── test_assortativity.py │ │ ├── test_augmentation.py │ │ ├── test_coalesce.py │ │ ├── test_convert.py │ │ ├── test_cross_entropy.py │ │ ├── test_degree.py │ │ ├── test_dropout.py │ │ ├── test_embedding.py │ │ ├── test_functions.py │ │ ├── test_geodesic.py │ │ ├── test_grid.py │ │ ├── test_hetero.py │ │ ├── test_homophily.py │ │ ├── test_index_sort.py │ │ ├── test_isolated.py │ │ ├── test_laplacian.py │ │ ├── test_lexsort.py │ │ ├── test_loop.py │ │ ├── test_map.py │ │ ├── test_mask.py │ │ ├── test_mesh_laplacian.py │ │ ├── test_negative_sampling.py │ │ ├── test_nested.py │ │ ├── test_noise_scheduler.py │ │ ├── test_normalize_edge_index.py │ │ ├── test_normalized_cut.py │ │ ├── test_num_nodes.py │ │ ├── test_one_hot.py │ │ ├── test_ppr.py │ │ ├── test_random.py │ │ ├── test_repeat.py │ │ ├── test_scatter.py │ │ ├── test_segment.py │ │ ├── test_select.py │ │ ├── test_smiles.py │ │ ├── test_softmax.py │ │ ├── test_sort_edge_index.py │ │ ├── test_sparse.py │ │ ├── test_spmm.py │ │ ├── test_subgraph.py │ │ ├── test_to_dense_adj.py │ │ ├── test_to_dense_batch.py │ │ ├── test_total_influence.py │ │ ├── test_train_test_split_edges.py │ │ ├── test_tree_decomposition.py │ │ ├── test_trim_to_layer.py │ │ ├── test_unbatch.py │ │ └── test_undirected.py │ └── visualization/ │ ├── test_graph_visualization.py │ └── test_influence.py └── torch_geometric/ ├── __init__.py ├── _compile.py ├── _onnx.py ├── backend.py ├── config_mixin.py ├── config_store.py ├── contrib/ │ ├── __init__.py │ ├── datasets/ │ │ └── __init__.py │ ├── explain/ │ │ ├── __init__.py │ │ └── pgm_explainer.py │ ├── nn/ │ │ ├── __init__.py │ │ ├── conv/ │ │ │ └── __init__.py │ │ └── models/ │ │ ├── __init__.py │ │ └── rbcd_attack.py │ └── transforms/ │ └── __init__.py ├── data/ │ ├── __init__.py │ ├── batch.py │ ├── collate.py │ ├── data.py │ ├── database.py │ ├── datapipes.py │ ├── dataset.py │ ├── download.py │ ├── extract.py │ ├── feature_store.py │ ├── graph_store.py │ ├── hetero_data.py │ ├── hypergraph_data.py │ ├── in_memory_dataset.py │ ├── lightning/ │ │ ├── __init__.py │ │ └── datamodule.py │ ├── makedirs.py │ ├── on_disk_dataset.py │ ├── remote_backend_utils.py │ ├── separate.py │ ├── storage.py │ ├── summary.py │ ├── temporal.py │ └── view.py ├── datasets/ │ ├── __init__.py │ ├── actor.py │ ├── airfrans.py │ ├── airports.py │ ├── amazon.py │ ├── amazon_book.py │ ├── amazon_products.py │ ├── aminer.py │ ├── aqsol.py │ ├── attributed_graph_dataset.py │ ├── ba2motif_dataset.py │ ├── ba_multi_shapes.py │ ├── ba_shapes.py │ ├── bitcoin_otc.py │ ├── brca_tgca.py │ ├── citation_full.py │ ├── city.py │ ├── coauthor.py │ ├── coma.py │ ├── cornell.py │ ├── dblp.py │ ├── dbp15k.py │ ├── deezer_europe.py │ ├── dgraph.py │ ├── dynamic_faust.py │ ├── elliptic.py │ ├── elliptic_temporal.py │ ├── email_eu_core.py │ ├── entities.py │ ├── explainer_dataset.py │ ├── facebook.py │ ├── fake.py │ ├── faust.py │ ├── flickr.py │ ├── freebase.py │ ├── gdelt.py │ ├── gdelt_lite.py │ ├── ged_dataset.py │ ├── gemsec.py │ ├── geometry.py │ ├── git_mol_dataset.py │ ├── github.py │ ├── gnn_benchmark_dataset.py │ ├── graph_generator/ │ │ ├── __init__.py │ │ ├── ba_graph.py │ │ ├── base.py │ │ ├── er_graph.py │ │ ├── grid_graph.py │ │ └── tree_graph.py │ ├── heterophilous_graph_dataset.py │ ├── hgb_dataset.py │ ├── hm.py │ ├── hydro_net.py │ ├── icews.py │ ├── igmc_dataset.py │ ├── imdb.py │ ├── infection_dataset.py │ ├── instruct_mol_dataset.py │ ├── jodie.py │ ├── karate.py │ ├── last_fm.py │ ├── lastfm_asia.py │ ├── linkx_dataset.py │ ├── lrgb.py │ ├── malnet_tiny.py │ ├── md17.py │ ├── medshapenet.py │ ├── mixhop_synthetic_dataset.py │ ├── mnist_superpixels.py │ ├── modelnet.py │ ├── molecule_gpt_dataset.py │ ├── molecule_net.py │ ├── motif_generator/ │ │ ├── __init__.py │ │ ├── base.py │ │ ├── custom.py │ │ ├── cycle.py │ │ ├── grid.py │ │ └── house.py │ ├── movie_lens.py │ ├── movie_lens_100k.py │ ├── movie_lens_1m.py │ ├── myket.py │ ├── nell.py │ ├── neurograph.py │ ├── ogb_mag.py │ ├── omdb.py │ ├── opf.py │ ├── ose_gvcs.py │ ├── particle.py │ ├── pascal.py │ ├── pascal_pf.py │ ├── pcpnet_dataset.py │ ├── pcqm4m.py │ ├── planetoid.py │ ├── polblogs.py │ ├── ppi.py │ ├── protein_mpnn_dataset.py │ ├── qm7.py │ ├── qm9.py │ ├── rcdd.py │ ├── reddit.py │ ├── reddit2.py │ ├── rel_link_pred_dataset.py │ ├── s3dis.py │ ├── sbm_dataset.py │ ├── shapenet.py │ ├── shrec2016.py │ ├── snap_dataset.py │ ├── suite_sparse.py │ ├── tag_dataset.py │ ├── taobao.py │ ├── teeth3ds.py │ ├── tosca.py │ ├── tu_dataset.py │ ├── twitch.py │ ├── upfd.py │ ├── utils/ │ │ ├── __init__.py │ │ └── cheatsheet.py │ ├── web_qsp_dataset.py │ ├── webkb.py │ ├── wikics.py │ ├── wikidata.py │ ├── wikipedia_network.py │ ├── willow_object_class.py │ ├── word_net.py │ ├── yelp.py │ └── zinc.py ├── debug.py ├── deprecation.py ├── device.py ├── distributed/ │ ├── __init__.py │ ├── dist_context.py │ ├── dist_link_neighbor_loader.py │ ├── dist_loader.py │ ├── dist_neighbor_loader.py │ ├── dist_neighbor_sampler.py │ ├── event_loop.py │ ├── local_feature_store.py │ ├── local_graph_store.py │ ├── partition.py │ ├── rpc.py │ └── utils.py ├── edge_index.py ├── experimental.py ├── explain/ │ ├── __init__.py │ ├── algorithm/ │ │ ├── __init__.py │ │ ├── attention_explainer.py │ │ ├── base.py │ │ ├── captum.py │ │ ├── captum_explainer.py │ │ ├── dummy_explainer.py │ │ ├── gnn_explainer.py │ │ ├── graphmask_explainer.py │ │ ├── pg_explainer.py │ │ └── utils.py │ ├── config.py │ ├── explainer.py │ ├── explanation.py │ └── metric/ │ ├── __init__.py │ ├── basic.py │ ├── faithfulness.py │ └── fidelity.py ├── graphgym/ │ ├── __init__.py │ ├── benchmark.py │ ├── checkpoint.py │ ├── cmd_args.py │ ├── config.py │ ├── contrib/ │ │ ├── __init__.py │ │ ├── act/ │ │ │ └── __init__.py │ │ ├── config/ │ │ │ └── __init__.py │ │ ├── encoder/ │ │ │ └── __init__.py │ │ ├── head/ │ │ │ └── __init__.py │ │ ├── layer/ │ │ │ ├── __init__.py │ │ │ └── generalconv.py │ │ ├── loader/ │ │ │ └── __init__.py │ │ ├── loss/ │ │ │ └── __init__.py │ │ ├── network/ │ │ │ └── __init__.py │ │ ├── optimizer/ │ │ │ └── __init__.py │ │ ├── pooling/ │ │ │ └── __init__.py │ │ ├── stage/ │ │ │ └── __init__.py │ │ ├── train/ │ │ │ └── __init__.py │ │ └── transform/ │ │ └── __init__.py │ ├── imports.py │ ├── init.py │ ├── loader.py │ ├── logger.py │ ├── loss.py │ ├── model_builder.py │ ├── models/ │ │ ├── __init__.py │ │ ├── act.py │ │ ├── encoder.py │ │ ├── gnn.py │ │ ├── head.py │ │ ├── layer.py │ │ ├── pooling.py │ │ └── transform.py │ ├── optim.py │ ├── register.py │ ├── train.py │ └── utils/ │ ├── LICENSE │ ├── __init__.py │ ├── agg_runs.py │ ├── comp_budget.py │ ├── device.py │ ├── epoch.py │ ├── io.py │ ├── plot.py │ └── tools.py ├── hash_tensor.py ├── home.py ├── index.py ├── inspector.py ├── io/ │ ├── __init__.py │ ├── fs.py │ ├── npz.py │ ├── obj.py │ ├── off.py │ ├── planetoid.py │ ├── ply.py │ ├── sdf.py │ ├── tu.py │ └── txt_array.py ├── isinstance.py ├── lazy_loader.py ├── llm/ │ ├── __init__.py │ ├── large_graph_indexer.py │ ├── models/ │ │ ├── __init__.py │ │ ├── g_retriever.py │ │ ├── git_mol.py │ │ ├── glem.py │ │ ├── llm.py │ │ ├── llm_judge.py │ │ ├── molecule_gpt.py │ │ ├── protein_mpnn.py │ │ ├── sentence_transformer.py │ │ ├── txt2kg.py │ │ └── vision_transformer.py │ ├── rag_loader.py │ └── utils/ │ ├── __init__.py │ ├── backend_utils.py │ ├── feature_store.py │ ├── graph_store.py │ └── vectorrag.py ├── loader/ │ ├── __init__.py │ ├── base.py │ ├── cache.py │ ├── cluster.py │ ├── data_list_loader.py │ ├── dataloader.py │ ├── dense_data_loader.py │ ├── dynamic_batch_sampler.py │ ├── graph_saint.py │ ├── hgt_loader.py │ ├── ibmb_loader.py │ ├── imbalanced_sampler.py │ ├── link_loader.py │ ├── link_neighbor_loader.py │ ├── mixin.py │ ├── neighbor_loader.py │ ├── neighbor_sampler.py │ ├── node_loader.py │ ├── prefetch.py │ ├── random_node_loader.py │ ├── shadow.py │ ├── temporal_dataloader.py │ ├── utils.py │ └── zip_loader.py ├── logging.py ├── metrics/ │ ├── __init__.py │ └── link_pred.py ├── nn/ │ ├── __init__.py │ ├── aggr/ │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── base.py │ │ ├── basic.py │ │ ├── deep_sets.py │ │ ├── equilibrium.py │ │ ├── fused.py │ │ ├── gmt.py │ │ ├── gru.py │ │ ├── lcm.py │ │ ├── lstm.py │ │ ├── mlp.py │ │ ├── multi.py │ │ ├── patch_transformer.py │ │ ├── quantile.py │ │ ├── scaler.py │ │ ├── set2set.py │ │ ├── set_transformer.py │ │ ├── sort.py │ │ ├── utils.py │ │ └── variance_preserving.py │ ├── attention/ │ │ ├── __init__.py │ │ ├── performer.py │ │ ├── polynormer.py │ │ ├── qformer.py │ │ └── sgformer.py │ ├── conv/ │ │ ├── __init__.py │ │ ├── agnn_conv.py │ │ ├── antisymmetric_conv.py │ │ ├── appnp.py │ │ ├── arma_conv.py │ │ ├── cg_conv.py │ │ ├── cheb_conv.py │ │ ├── cluster_gcn_conv.py │ │ ├── collect.jinja │ │ ├── cugraph/ │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── gat_conv.py │ │ │ ├── rgcn_conv.py │ │ │ └── sage_conv.py │ │ ├── dir_gnn_conv.py │ │ ├── dna_conv.py │ │ ├── edge_conv.py │ │ ├── edge_updater.jinja │ │ ├── eg_conv.py │ │ ├── fa_conv.py │ │ ├── feast_conv.py │ │ ├── film_conv.py │ │ ├── fused_gat_conv.py │ │ ├── gat_conv.py │ │ ├── gated_graph_conv.py │ │ ├── gatv2_conv.py │ │ ├── gcn2_conv.py │ │ ├── gcn_conv.py │ │ ├── gen_conv.py │ │ ├── general_conv.py │ │ ├── gin_conv.py │ │ ├── gmm_conv.py │ │ ├── gps_conv.py │ │ ├── graph_conv.py │ │ ├── gravnet_conv.py │ │ ├── han_conv.py │ │ ├── heat_conv.py │ │ ├── hetero_conv.py │ │ ├── hgt_conv.py │ │ ├── hypergraph_conv.py │ │ ├── le_conv.py │ │ ├── lg_conv.py │ │ ├── meshcnn_conv.py │ │ ├── message_passing.py │ │ ├── mf_conv.py │ │ ├── mixhop_conv.py │ │ ├── nn_conv.py │ │ ├── pan_conv.py │ │ ├── pdn_conv.py │ │ ├── pna_conv.py │ │ ├── point_conv.py │ │ ├── point_gnn_conv.py │ │ ├── point_transformer_conv.py │ │ ├── ppf_conv.py │ │ ├── propagate.jinja │ │ ├── res_gated_graph_conv.py │ │ ├── rgat_conv.py │ │ ├── rgcn_conv.py │ │ ├── sage_conv.py │ │ ├── sg_conv.py │ │ ├── signed_conv.py │ │ ├── simple_conv.py │ │ ├── spline_conv.py │ │ ├── ssg_conv.py │ │ ├── supergat_conv.py │ │ ├── tag_conv.py │ │ ├── transformer_conv.py │ │ ├── utils/ │ │ │ ├── __init__.py │ │ │ └── cheatsheet.py │ │ ├── wl_conv.py │ │ ├── wl_conv_continuous.py │ │ └── x_conv.py │ ├── data_parallel.py │ ├── dense/ │ │ ├── __init__.py │ │ ├── dense_gat_conv.py │ │ ├── dense_gcn_conv.py │ │ ├── dense_gin_conv.py │ │ ├── dense_graph_conv.py │ │ ├── dense_sage_conv.py │ │ ├── diff_pool.py │ │ ├── dmon_pool.py │ │ ├── linear.py │ │ └── mincut_pool.py │ ├── encoding.py │ ├── functional/ │ │ ├── __init__.py │ │ ├── bro.py │ │ └── gini.py │ ├── fx.py │ ├── glob.py │ ├── inits.py │ ├── kge/ │ │ ├── __init__.py │ │ ├── base.py │ │ ├── complex.py │ │ ├── distmult.py │ │ ├── loader.py │ │ ├── rotate.py │ │ └── transe.py │ ├── lr_scheduler.py │ ├── model_hub.py │ ├── models/ │ │ ├── __init__.py │ │ ├── attentive_fp.py │ │ ├── attract_repel.py │ │ ├── autoencoder.py │ │ ├── basic_gnn.py │ │ ├── captum.py │ │ ├── correct_and_smooth.py │ │ ├── deep_graph_infomax.py │ │ ├── deepgcn.py │ │ ├── dimenet.py │ │ ├── dimenet_utils.py │ │ ├── gnnff.py │ │ ├── gpse.py │ │ ├── graph_mixer.py │ │ ├── graph_unet.py │ │ ├── jumping_knowledge.py │ │ ├── label_prop.py │ │ ├── lightgcn.py │ │ ├── linkx.py │ │ ├── lpformer.py │ │ ├── mask_label.py │ │ ├── meta.py │ │ ├── metapath2vec.py │ │ ├── mlp.py │ │ ├── neural_fingerprint.py │ │ ├── node2vec.py │ │ ├── pmlp.py │ │ ├── polynormer.py │ │ ├── re_net.py │ │ ├── rect.py │ │ ├── rev_gnn.py │ │ ├── schnet.py │ │ ├── sgformer.py │ │ ├── signed_gcn.py │ │ ├── tgn.py │ │ └── visnet.py │ ├── module_dict.py │ ├── norm/ │ │ ├── __init__.py │ │ ├── batch_norm.py │ │ ├── diff_group_norm.py │ │ ├── graph_norm.py │ │ ├── graph_size_norm.py │ │ ├── instance_norm.py │ │ ├── layer_norm.py │ │ ├── mean_subtraction_norm.py │ │ ├── msg_norm.py │ │ └── pair_norm.py │ ├── parameter_dict.py │ ├── pool/ │ │ ├── __init__.py │ │ ├── approx_knn.py │ │ ├── asap.py │ │ ├── avg_pool.py │ │ ├── cluster_pool.py │ │ ├── connect/ │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ └── filter_edges.py │ │ ├── consecutive.py │ │ ├── decimation.py │ │ ├── edge_pool.py │ │ ├── glob.py │ │ ├── graclus.py │ │ ├── knn.py │ │ ├── max_pool.py │ │ ├── mem_pool.py │ │ ├── pan_pool.py │ │ ├── pool.py │ │ ├── sag_pool.py │ │ ├── select/ │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ └── topk.py │ │ ├── topk_pool.py │ │ └── voxel_grid.py │ ├── reshape.py │ ├── resolver.py │ ├── sequential.jinja │ ├── sequential.py │ ├── summary.py │ ├── to_fixed_size_transformer.py │ ├── to_hetero_module.py │ ├── to_hetero_transformer.py │ ├── to_hetero_with_bases_transformer.py │ └── unpool/ │ ├── __init__.py │ └── knn_interpolate.py ├── profile/ │ ├── __init__.py │ ├── benchmark.py │ ├── nvtx.py │ ├── profile.py │ ├── profiler.py │ └── utils.py ├── resolver.py ├── sampler/ │ ├── __init__.py │ ├── base.py │ ├── hgt_sampler.py │ ├── neighbor_sampler.py │ └── utils.py ├── seed.py ├── template.py ├── testing/ │ ├── __init__.py │ ├── asserts.py │ ├── data.py │ ├── decorators.py │ ├── distributed.py │ ├── feature_store.py │ └── graph_store.py ├── transforms/ │ ├── __init__.py │ ├── add_gpse.py │ ├── add_metapaths.py │ ├── add_positional_encoding.py │ ├── add_remaining_self_loops.py │ ├── add_self_loops.py │ ├── base_transform.py │ ├── cartesian.py │ ├── center.py │ ├── compose.py │ ├── constant.py │ ├── delaunay.py │ ├── distance.py │ ├── face_to_edge.py │ ├── feature_propagation.py │ ├── fixed_points.py │ ├── gcn_norm.py │ ├── gdc.py │ ├── generate_mesh_normals.py │ ├── grid_sampling.py │ ├── half_hop.py │ ├── knn_graph.py │ ├── laplacian_lambda_max.py │ ├── largest_connected_components.py │ ├── line_graph.py │ ├── linear_transformation.py │ ├── local_cartesian.py │ ├── local_degree_profile.py │ ├── mask.py │ ├── node_property_split.py │ ├── normalize_features.py │ ├── normalize_rotation.py │ ├── normalize_scale.py │ ├── one_hot_degree.py │ ├── pad.py │ ├── point_pair_features.py │ ├── polar.py │ ├── radius_graph.py │ ├── random_flip.py │ ├── random_jitter.py │ ├── random_link_split.py │ ├── random_node_split.py │ ├── random_rotate.py │ ├── random_scale.py │ ├── random_shear.py │ ├── remove_duplicated_edges.py │ ├── remove_isolated_nodes.py │ ├── remove_self_loops.py │ ├── remove_training_classes.py │ ├── rooted_subgraph.py │ ├── sample_points.py │ ├── sign.py │ ├── spherical.py │ ├── svd_feature_reduction.py │ ├── target_indegree.py │ ├── to_dense.py │ ├── to_device.py │ ├── to_sparse_tensor.py │ ├── to_superpixels.py │ ├── to_undirected.py │ ├── two_hop.py │ └── virtual_node.py ├── typing.py ├── utils/ │ ├── __init__.py │ ├── _assortativity.py │ ├── _coalesce.py │ ├── _degree.py │ ├── _grid.py │ ├── _homophily.py │ ├── _index_sort.py │ ├── _lexsort.py │ ├── _negative_sampling.py │ ├── _normalize_edge_index.py │ ├── _normalized_cut.py │ ├── _one_hot.py │ ├── _scatter.py │ ├── _segment.py │ ├── _select.py │ ├── _softmax.py │ ├── _sort_edge_index.py │ ├── _spmm.py │ ├── _subgraph.py │ ├── _to_dense_adj.py │ ├── _to_dense_batch.py │ ├── _train_test_split_edges.py │ ├── _tree_decomposition.py │ ├── _trim_to_layer.py │ ├── _unbatch.py │ ├── augmentation.py │ ├── convert.py │ ├── cross_entropy.py │ ├── dropout.py │ ├── embedding.py │ ├── functions.py │ ├── geodesic.py │ ├── hetero.py │ ├── influence.py │ ├── isolated.py │ ├── laplacian.py │ ├── loop.py │ ├── map.py │ ├── mask.py │ ├── mesh_laplacian.py │ ├── mixin.py │ ├── nested.py │ ├── noise_scheduler.py │ ├── num_nodes.py │ ├── ppr.py │ ├── random.py │ ├── repeat.py │ ├── smiles.py │ ├── sparse.py │ └── undirected.py ├── visualization/ │ ├── __init__.py │ ├── graph.py │ └── influence.py └── warnings.py