LICENSE
README.md
pyproject.toml
setup.py
k3_node/__init__.py
k3_node/metrics.py
k3_node/test_docstring_examples.py
k3_node/test_training_forwarding.py
k3_node/training.py
k3_node.egg-info/PKG-INFO
k3_node.egg-info/SOURCES.txt
k3_node.egg-info/dependency_links.txt
k3_node.egg-info/requires.txt
k3_node.egg-info/top_level.txt
k3_node/applications/__init__.py
k3_node/applications/bio/__init__.py
k3_node/applications/chemistry/__init__.py
k3_node/applications/materials/__init__.py
k3_node/applications/materials/basis.py
k3_node/applications/materials/chgnet.py
k3_node/applications/materials/core.py
k3_node/applications/materials/grace.py
k3_node/applications/materials/io.py
k3_node/applications/materials/m3gnet.py
k3_node/applications/materials/megnet.py
k3_node/applications/materials/qet.py
k3_node/applications/materials/readout.py
k3_node/applications/materials/so3net.py
k3_node/applications/materials/tensornet.py
k3_node/applications/materials/test_materials.py
k3_node/applications/materials/wrappers.py
k3_node/data/__init__.py
k3_node/data/batch.py
k3_node/data/collate.py
k3_node/data/data.py
k3_node/data/database.py
k3_node/data/dataset.py
k3_node/data/download.py
k3_node/data/extract.py
k3_node/data/feature_store.py
k3_node/data/graph_store.py
k3_node/data/hetero_data.py
k3_node/data/hypergraph_data.py
k3_node/data/in_memory_dataset.py
k3_node/data/makedirs.py
k3_node/data/on_disk_dataset.py
k3_node/data/separate.py
k3_node/data/storage.py
k3_node/data/temporal.py
k3_node/data/test_batch.py
k3_node/data/test_data.py
k3_node/data/test_dataset_and_stores.py
k3_node/data/test_hetero_data.py
k3_node/data/test_temporal_and_hyper.py
k3_node/data/view.py
k3_node/datasets/__init__.py
k3_node/datasets/actor.py
k3_node/datasets/airports.py
k3_node/datasets/amazon.py
k3_node/datasets/ba2motif_dataset.py
k3_node/datasets/ba_shapes.py
k3_node/datasets/bitcoin_otc.py
k3_node/datasets/citation_full.py
k3_node/datasets/coauthor.py
k3_node/datasets/dblp.py
k3_node/datasets/digits.py
k3_node/datasets/email_eu_core.py
k3_node/datasets/entities.py
k3_node/datasets/explainer_dataset.py
k3_node/datasets/facebook.py
k3_node/datasets/fake.py
k3_node/datasets/freebase.py
k3_node/datasets/geometric_shapes.py
k3_node/datasets/github.py
k3_node/datasets/icews.py
k3_node/datasets/imdb.py
k3_node/datasets/jodie.py
k3_node/datasets/karate.py
k3_node/datasets/lastfm_asia.py
k3_node/datasets/mesh_correspondence.py
k3_node/datasets/molecule_net.py
k3_node/datasets/movielens.py
k3_node/datasets/planetoid.py
k3_node/datasets/polblogs.py
k3_node/datasets/ppi.py
k3_node/datasets/qm7.py
k3_node/datasets/qm9.py
k3_node/datasets/reddit.py
k3_node/datasets/sbm_dataset.py
k3_node/datasets/seal.py
k3_node/datasets/shape_scenes.py
k3_node/datasets/test_datasets.py
k3_node/datasets/tu_dataset.py
k3_node/datasets/twitch.py
k3_node/datasets/webkb.py
k3_node/datasets/wikics.py
k3_node/datasets/word_net.py
k3_node/datasets/graph_generator/__init__.py
k3_node/datasets/graph_generator/ba_graph.py
k3_node/datasets/graph_generator/base.py
k3_node/datasets/graph_generator/er_graph.py
k3_node/datasets/motif_generator/__init__.py
k3_node/datasets/motif_generator/base.py
k3_node/datasets/motif_generator/custom.py
k3_node/datasets/motif_generator/cycle.py
k3_node/datasets/motif_generator/house.py
k3_node/etl/__init__.py
k3_node/etl/encoders.py
k3_node/etl/graph_builders.py
k3_node/etl/relational_to_graph.py
k3_node/etl/table_to_graph.py
k3_node/etl/test_etl.py
k3_node/export/__init__.py
k3_node/export/cross_backend.py
k3_node/export/onnx_exporter.py
k3_node/export/runtime.py
k3_node/export/tensorrt_exporter.py
k3_node/export/test_export.py
k3_node/export/tflite_exporter.py
k3_node/hub/__init__.py
k3_node/hub/dataset_hub.py
k3_node/hub/hub_mixin.py
k3_node/hub/model_card.py
k3_node/hub/test_hub.py
k3_node/io/__init__.py
k3_node/io/fs.py
k3_node/io/npz.py
k3_node/io/off.py
k3_node/io/planetoid.py
k3_node/io/tu.py
k3_node/io/txt_array.py
k3_node/layers/__init__.py
k3_node/layers/aggr/__init__.py
k3_node/layers/aggr/attention.py
k3_node/layers/aggr/base.py
k3_node/layers/aggr/basic.py
k3_node/layers/aggr/deep_sets.py
k3_node/layers/aggr/deepsets.py
k3_node/layers/aggr/equilibrium.py
k3_node/layers/aggr/fused.py
k3_node/layers/aggr/gmt.py
k3_node/layers/aggr/gru.py
k3_node/layers/aggr/lcm.py
k3_node/layers/aggr/lstm.py
k3_node/layers/aggr/mlp.py
k3_node/layers/aggr/multi.py
k3_node/layers/aggr/patch_transformer.py
k3_node/layers/aggr/quantile.py
k3_node/layers/aggr/resolver.py
k3_node/layers/aggr/scaler.py
k3_node/layers/aggr/set2set.py
k3_node/layers/aggr/set_transformer.py
k3_node/layers/aggr/sort.py
k3_node/layers/aggr/test_aggr.py
k3_node/layers/aggr/utils.py
k3_node/layers/aggr/variance_preserving.py
k3_node/layers/attention/__init__.py
k3_node/layers/attention/pair_attention.py
k3_node/layers/attention/performer.py
k3_node/layers/attention/polynormer.py
k3_node/layers/attention/qformer.py
k3_node/layers/attention/sgformer.py
k3_node/layers/attention/test_attention.py
k3_node/layers/attention/test_pair_attention.py
k3_node/layers/conv/__init__.py
k3_node/layers/conv/agnn_conv.py
k3_node/layers/conv/antisymmetric_conv.py
k3_node/layers/conv/appnp.py
k3_node/layers/conv/appnp_conv.py
k3_node/layers/conv/arma_conv.py
k3_node/layers/conv/cg_conv.py
k3_node/layers/conv/cheb_conv.py
k3_node/layers/conv/cluster_gcn_conv.py
k3_node/layers/conv/conv.py
k3_node/layers/conv/crystal_conv.py
k3_node/layers/conv/cugraph.py
k3_node/layers/conv/diffusion_conv.py
k3_node/layers/conv/dir_gnn_conv.py
k3_node/layers/conv/dna_conv.py
k3_node/layers/conv/edge_conv.py
k3_node/layers/conv/eg_conv.py
k3_node/layers/conv/fa_conv.py
k3_node/layers/conv/feast_conv.py
k3_node/layers/conv/film_conv.py
k3_node/layers/conv/gat_conv.py
k3_node/layers/conv/gated_graph_conv.py
k3_node/layers/conv/gatv2_conv.py
k3_node/layers/conv/gcn.py
k3_node/layers/conv/gcn2_conv.py
k3_node/layers/conv/gcn_conv.py
k3_node/layers/conv/gen_conv.py
k3_node/layers/conv/general_conv.py
k3_node/layers/conv/gin_conv.py
k3_node/layers/conv/gmm_conv.py
k3_node/layers/conv/gps_conv.py
k3_node/layers/conv/graph_attention.py
k3_node/layers/conv/graph_conv.py
k3_node/layers/conv/gravnet_conv.py
k3_node/layers/conv/han_conv.py
k3_node/layers/conv/heat_conv.py
k3_node/layers/conv/hetero_conv.py
k3_node/layers/conv/hgt_conv.py
k3_node/layers/conv/hypergraph_conv.py
k3_node/layers/conv/le_conv.py
k3_node/layers/conv/lg_conv.py
k3_node/layers/conv/meshcnn_conv.py
k3_node/layers/conv/message_passing.py
k3_node/layers/conv/mf_conv.py
k3_node/layers/conv/mixhop_conv.py
k3_node/layers/conv/nn_conv.py
k3_node/layers/conv/pan_conv.py
k3_node/layers/conv/pdn_conv.py
k3_node/layers/conv/pna_conv.py
k3_node/layers/conv/point_conv.py
k3_node/layers/conv/point_gnn_conv.py
k3_node/layers/conv/point_transformer_conv.py
k3_node/layers/conv/ppf_conv.py
k3_node/layers/conv/ppnp.py
k3_node/layers/conv/res_gated_graph_conv.py
k3_node/layers/conv/rgat_conv.py
k3_node/layers/conv/rgcn_conv.py
k3_node/layers/conv/sage_conv.py
k3_node/layers/conv/sg_conv.py
k3_node/layers/conv/signed_conv.py
k3_node/layers/conv/simple_conv.py
k3_node/layers/conv/spline_conv.py
k3_node/layers/conv/ssg_conv.py
k3_node/layers/conv/supergat_conv.py
k3_node/layers/conv/tag_conv.py
k3_node/layers/conv/test_backend_consistency.py
k3_node/layers/conv/test_conv.py
k3_node/layers/conv/test_conv_pyg.py
k3_node/layers/conv/transformer_conv.py
k3_node/layers/conv/utils.py
k3_node/layers/conv/wl_conv.py
k3_node/layers/conv/x_conv.py
k3_node/layers/dense/__init__.py
k3_node/layers/dense/dense_gat_conv.py
k3_node/layers/dense/dense_gcn_conv.py
k3_node/layers/dense/dense_gin_conv.py
k3_node/layers/dense/dense_graph_conv.py
k3_node/layers/dense/dense_sage_conv.py
k3_node/layers/dense/diff_pool.py
k3_node/layers/dense/dmon_pool.py
k3_node/layers/dense/linear.py
k3_node/layers/dense/mincut_pool.py
k3_node/layers/dense/test_dense.py
k3_node/layers/functional/__init__.py
k3_node/layers/functional/bro.py
k3_node/layers/functional/edge_dropout.py
k3_node/layers/functional/gini.py
k3_node/layers/functional/test_functional.py
k3_node/layers/kge/__init__.py
k3_node/layers/kge/base.py
k3_node/layers/kge/complex.py
k3_node/layers/kge/distmult.py
k3_node/layers/kge/loader.py
k3_node/layers/kge/rotate.py
k3_node/layers/kge/test_kge.py
k3_node/layers/kge/transe.py
k3_node/layers/norm/__init__.py
k3_node/layers/norm/batch_norm.py
k3_node/layers/norm/diff_group_norm.py
k3_node/layers/norm/graph_norm.py
k3_node/layers/norm/graph_size_norm.py
k3_node/layers/norm/instance_norm.py
k3_node/layers/norm/layer_norm.py
k3_node/layers/norm/mean_subtraction_norm.py
k3_node/layers/norm/msg_norm.py
k3_node/layers/norm/pair_norm.py
k3_node/layers/norm/test_norm.py
k3_node/layers/pool/__init__.py
k3_node/layers/pool/approx_knn.py
k3_node/layers/pool/asap.py
k3_node/layers/pool/avg_pool.py
k3_node/layers/pool/cluster_pool.py
k3_node/layers/pool/consecutive.py
k3_node/layers/pool/decimation.py
k3_node/layers/pool/edge_pool.py
k3_node/layers/pool/glob.py
k3_node/layers/pool/graclus.py
k3_node/layers/pool/knn.py
k3_node/layers/pool/max_pool.py
k3_node/layers/pool/mem_pool.py
k3_node/layers/pool/pan_pool.py
k3_node/layers/pool/point_cloud.py
k3_node/layers/pool/pool.py
k3_node/layers/pool/sag_pool.py
k3_node/layers/pool/test_pool.py
k3_node/layers/pool/topk_pool.py
k3_node/layers/pool/voxel_grid.py
k3_node/layers/pool/connect/__init__.py
k3_node/layers/pool/connect/base.py
k3_node/layers/pool/connect/filter_edges.py
k3_node/layers/pool/select/__init__.py
k3_node/layers/pool/select/base.py
k3_node/layers/pool/select/topk.py
k3_node/layers/unpool/__init__.py
k3_node/layers/unpool/knn_interpolate.py
k3_node/layers/unpool/test_unpool.py
k3_node/loader/__init__.py
k3_node/loader/base.py
k3_node/loader/cache.py
k3_node/loader/cluster.py
k3_node/loader/data_list_loader.py
k3_node/loader/dataloader.py
k3_node/loader/dense_data_loader.py
k3_node/loader/dynamic_batch_sampler.py
k3_node/loader/graph_saint.py
k3_node/loader/hgt_loader.py
k3_node/loader/imbalanced_sampler.py
k3_node/loader/keras_dataset.py
k3_node/loader/link_loader.py
k3_node/loader/link_neighbor_loader.py
k3_node/loader/mixin.py
k3_node/loader/neighbor_loader.py
k3_node/loader/neighbor_sampler.py
k3_node/loader/node_loader.py
k3_node/loader/prefetch.py
k3_node/loader/random_node_loader.py
k3_node/loader/sampler_utils.py
k3_node/loader/shadow.py
k3_node/loader/temporal_dataloader.py
k3_node/loader/test_dataloader.py
k3_node/loader/test_keras_dataset.py
k3_node/loader/test_neighbor_loader.py
k3_node/loader/test_sampler_utils.py
k3_node/loader/test_samplers.py
k3_node/loader/test_subgraph_loaders.py
k3_node/loader/utils.py
k3_node/loader/zip_loader.py
k3_node/models/__init__.py
k3_node/models/attentive_fp.py
k3_node/models/attract_repel.py
k3_node/models/autoencoder.py
k3_node/models/basic_gnn.py
k3_node/models/captum.py
k3_node/models/correct_and_smooth.py
k3_node/models/deep_graph_infomax.py
k3_node/models/deepgcn.py
k3_node/models/dimenet.py
k3_node/models/dimenet_utils.py
k3_node/models/gnnff.py
k3_node/models/gps_model.py
k3_node/models/gpse.py
k3_node/models/graph_unet.py
k3_node/models/graphmae2.py
k3_node/models/graphormer.py
k3_node/models/graphormer_3d.py
k3_node/models/grover.py
k3_node/models/jumping_knowledge.py
k3_node/models/label_prop.py
k3_node/models/lightgcn.py
k3_node/models/linkx.py
k3_node/models/lpformer.py
k3_node/models/mask_label.py
k3_node/models/meta.py
k3_node/models/metapath2vec.py
k3_node/models/mlp.py
k3_node/models/mole_bert.py
k3_node/models/neural_fingerprint.py
k3_node/models/node2vec.py
k3_node/models/pmlp.py
k3_node/models/polynormer.py
k3_node/models/rect.py
k3_node/models/renet.py
k3_node/models/rev_gnn.py
k3_node/models/schnet.py
k3_node/models/sgformer.py
k3_node/models/signed_gcn.py
k3_node/models/test_attentive_fp.py
k3_node/models/test_attract_repel.py
k3_node/models/test_autoencoder.py
k3_node/models/test_basic_gnn.py
k3_node/models/test_correct_and_smooth.py
k3_node/models/test_deep_graph_infomax.py
k3_node/models/test_deepgcn.py
k3_node/models/test_dimenet.py
k3_node/models/test_domain_apis.py
k3_node/models/test_gnnff.py
k3_node/models/test_gps_model.py
k3_node/models/test_gpse.py
k3_node/models/test_graph_unet.py
k3_node/models/test_graphmae2.py
k3_node/models/test_graphormer.py
k3_node/models/test_graphormer3d.py
k3_node/models/test_grover.py
k3_node/models/test_jumping_knowledge.py
k3_node/models/test_label_prop.py
k3_node/models/test_lightgcn.py
k3_node/models/test_linkx.py
k3_node/models/test_lpformer.py
k3_node/models/test_mask_label.py
k3_node/models/test_meta.py
k3_node/models/test_metapath2vec.py
k3_node/models/test_mlp.py
k3_node/models/test_mole_bert.py
k3_node/models/test_neural_fingerprint.py
k3_node/models/test_node2vec.py
k3_node/models/test_pmlp.py
k3_node/models/test_polynormer.py
k3_node/models/test_rect.py
k3_node/models/test_renet.py
k3_node/models/test_rev_gnn.py
k3_node/models/test_schnet.py
k3_node/models/test_sgformer.py
k3_node/models/test_signed_gcn.py
k3_node/models/test_tgn.py
k3_node/models/test_unimol.py
k3_node/models/test_unimol2.py
k3_node/models/test_unimol_plus.py
k3_node/models/test_visnet.py
k3_node/models/tgn.py
k3_node/models/unimol.py
k3_node/models/unimol2.py
k3_node/models/unimol_docking_v2.py
k3_node/models/unimol_plus.py
k3_node/models/utils.py
k3_node/models/visnet.py
k3_node/models/bio/__init__.py
k3_node/models/chemistry/__init__.py
k3_node/models/materials/__init__.py
k3_node/ops/__init__.py
k3_node/ops/conv.py
k3_node/ops/creation.py
k3_node/ops/graph.py
k3_node/ops/host.py
k3_node/ops/matmul.py
k3_node/ops/numpy.py
k3_node/ops/segment.py
k3_node/ops/sparse.py
k3_node/rag/__init__.py
k3_node/rag/encoders.py
k3_node/rag/pipeline.py
k3_node/rag/projector.py
k3_node/rag/subgraph.py
k3_node/rag/test_rag.py
k3_node/rag/verbalizer.py
k3_node/tasks/__init__.py
k3_node/tasks/backbone_resolver.py
k3_node/tasks/base.py
k3_node/tasks/graph_classification.py
k3_node/tasks/graph_regression.py
k3_node/tasks/link_prediction.py
k3_node/tasks/node_classification.py
k3_node/tasks/node_regression.py
k3_node/tasks/test_tasks.py
k3_node/transforms/__init__.py
k3_node/transforms/base_transform.py
k3_node/transforms/compose.py
k3_node/transforms/general.py
k3_node/transforms/graph.py
k3_node/transforms/spatial.py
k3_node/transforms/test_random_link_split.py
k3_node/transforms/test_spatial_transforms.py
k3_node/transforms/test_transforms.py
k3_node/transforms/utils.py
k3_node/utils/__init__.py
k3_node/utils/backend_import.py
k3_node/utils/graph.py
k3_node/utils/keras.py
k3_node/utils/random.py
k3_node/utils/smiles.py