All modules for which code is available

• gigl.analytics.graph_validation.bq_graph_validator
• gigl.common.beam.better_tfrecordio
• gigl.common.beam.coders
• gigl.common.collections.frozen_dict
• gigl.common.collections.itertools
• gigl.common.collections.sorted_dict
• gigl.common.constants
• gigl.common.data.dataloaders
• gigl.common.data.export
• gigl.common.data.load_torch_tensors
• gigl.common.env_config
• gigl.common.logger
• gigl.common.metrics.base_metrics
• gigl.common.metrics.decorators
• gigl.common.metrics.metrics_interface
• gigl.common.services.dataproc
• gigl.common.services.gcp_cloud_logging
• gigl.common.services.vertex_ai
• gigl.common.types.resource_config
• gigl.common.types.uri.gcs_uri
• gigl.common.types.uri.http_uri
• gigl.common.types.uri.local_uri
• gigl.common.types.uri.uri
• gigl.common.types.uri.uri_factory
• gigl.common.types.wrappers.kfp_api
• gigl.common.utils.compute.random
• gigl.common.utils.compute.serialization.coder
• gigl.common.utils.compute.serialization.serialize_np
• gigl.common.utils.compute.serialization.serialize_protos
• gigl.common.utils.compute.serialization.serialize_string
• gigl.common.utils.dataflow
• gigl.common.utils.decorator
• gigl.common.utils.func_tools
• gigl.common.utils.gcs
• gigl.common.utils.jupyter_magics
• gigl.common.utils.kfp
• gigl.common.utils.local_fs
• gigl.common.utils.os_utils
• gigl.common.utils.proto_utils
• gigl.common.utils.retry
• gigl.common.utils.tensorflow_schema
• gigl.common.utils.test_utils
• gigl.common.utils.torch_training
• gigl.common.utils.vertex_ai_context
• gigl.distributed.constants
• gigl.distributed.dataset_factory
• gigl.distributed.dist_context
• gigl.distributed.dist_link_prediction_dataset
• gigl.distributed.dist_partitioner
• gigl.distributed.dist_range_partitioner
• gigl.distributed.distributed_neighborloader
• gigl.distributed.utils.device
• gigl.distributed.utils.init_neighbor_loader_worker
• gigl.distributed.utils.partition_book
• gigl.distributed.utils.serialized_graph_metadata_translator
• gigl.env.dep_constants
• gigl.env.pipelines_config
• gigl.orchestration.kubeflow.kfp_orchestrator
• gigl.orchestration.kubeflow.kfp_pipeline
• gigl.orchestration.kubeflow.runner
• gigl.orchestration.kubeflow.utils.glt_backend
• gigl.orchestration.kubeflow.utils.log_metrics
• gigl.orchestration.kubeflow.utils.resource
• gigl.orchestration.local.runner
• gigl.src.applied_tasks.test_tasks.academic
• gigl.src.common.constants.bq
• gigl.src.common.constants.components
• gigl.src.common.constants.gcs
• gigl.src.common.constants.graph_metadata
• gigl.src.common.constants.local_fs
• gigl.src.common.constants.metrics
• gigl.src.common.constants.resource_config
• gigl.src.common.constants.time
• gigl.src.common.constants.training
• gigl.src.common.graph_builder.abstract_graph_builder
• gigl.src.common.graph_builder.gbml_graph_protocol
• gigl.src.common.graph_builder.graph_builder_factory
• gigl.src.common.graph_builder.pyg_graph_builder
• gigl.src.common.graph_builder.pyg_graph_data
• gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec
• gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec
• gigl.src.common.modeling_task_specs.node_classification_modeling_task_spec
• gigl.src.common.modeling_task_specs.utils.early_stop
• gigl.src.common.modeling_task_specs.utils.infer
• gigl.src.common.modeling_task_specs.utils.parameter_loading
• gigl.src.common.modeling_task_specs.utils.profiler_wrapper
• gigl.src.common.models.layers.count_min_sketch
• gigl.src.common.models.layers.decoder
• gigl.src.common.models.layers.feature_interaction
• gigl.src.common.models.layers.loss
• gigl.src.common.models.layers.normalization
• gigl.src.common.models.layers.task
• gigl.src.common.models.pyg.graph.augmentations
• gigl.src.common.models.pyg.heterogeneous
• gigl.src.common.models.pyg.homogeneous
• gigl.src.common.models.pyg.link_prediction
• gigl.src.common.models.pyg.nn.conv.edge_attr_gat_conv
• gigl.src.common.models.pyg.nn.conv.gin_conv
• gigl.src.common.models.pyg.nn.conv.hgt_conv
• gigl.src.common.models.pyg.nn.conv.simplehgn_conv
• gigl.src.common.models.pyg.nn.models.feature_embedding
• gigl.src.common.models.pyg.nn.models.feature_interaction
• gigl.src.common.models.pyg.nn.models.jumping_knowledge
• gigl.src.common.models.pyg.utils
• gigl.src.common.models.utils.torch
• gigl.src.common.translators.gbml_protos_translator
• gigl.src.common.translators.model_eval_metrics_translator
• gigl.src.common.translators.training_samples_protos_translator
• gigl.src.common.types
• gigl.src.common.types.dataflow_job_options
• gigl.src.common.types.dataset_split
• gigl.src.common.types.exception
• gigl.src.common.types.features
• gigl.src.common.types.graph_data
• gigl.src.common.types.model
• gigl.src.common.types.model_eval_metrics
• gigl.src.common.types.pb_wrappers.dataset_metadata
• gigl.src.common.types.pb_wrappers.dataset_metadata_utils
• gigl.src.common.types.pb_wrappers.flattened_graph_metadata
• gigl.src.common.types.pb_wrappers.gbml_config
• gigl.src.common.types.pb_wrappers.gigl_resource_config
• gigl.src.common.types.pb_wrappers.graph_data_types
• gigl.src.common.types.pb_wrappers.graph_data_types_utils
• gigl.src.common.types.pb_wrappers.graph_metadata
• gigl.src.common.types.pb_wrappers.preprocessed_metadata
• gigl.src.common.types.pb_wrappers.sampling_op
• gigl.src.common.types.pb_wrappers.subgraph_sampling_strategy
• gigl.src.common.types.pb_wrappers.task_metadata
• gigl.src.common.types.pb_wrappers.trained_model_metadata
• gigl.src.common.types.pb_wrappers.types
• gigl.src.common.types.task_inputs
• gigl.src.common.types.task_metadata
• gigl.src.common.utils.bq
• gigl.src.common.utils.data.feature_serialization
• gigl.src.common.utils.data.training
• gigl.src.common.utils.dataflow
• gigl.src.common.utils.eval_metrics
• gigl.src.common.utils.file_loader
• gigl.src.common.utils.gbml_protos
• gigl.src.common.utils.metrics_service_provider
• gigl.src.common.utils.model
• gigl.src.common.utils.scheduler
• gigl.src.common.utils.spark_job_manager
• gigl.src.common.utils.time
• gigl.src.common.utils.timeout
• gigl.src.config_populator.config_populator
• gigl.src.data_preprocessor.data_preprocessor
• gigl.src.data_preprocessor.lib.data_preprocessor_config
• gigl.src.data_preprocessor.lib.enumerate.queries
• gigl.src.data_preprocessor.lib.enumerate.utils
• gigl.src.data_preprocessor.lib.ingest.bigquery
• gigl.src.data_preprocessor.lib.ingest.reference
• gigl.src.data_preprocessor.lib.transform.tf_value_encoder
• gigl.src.data_preprocessor.lib.transform.transformed_features_info
• gigl.src.data_preprocessor.lib.transform.utils
• gigl.src.data_preprocessor.lib.types
• gigl.src.inference.inferencer
• gigl.src.inference.lib.assets
• gigl.src.inference.v1.gnn_inferencer
• gigl.src.inference.v1.lib.base_inference_blueprint
• gigl.src.inference.v1.lib.base_inferencer
• gigl.src.inference.v1.lib.inference_blueprint_factory
• gigl.src.inference.v1.lib.inference_output_schema
• gigl.src.inference.v1.lib.node_anchor_based_link_prediction_inferencer
• gigl.src.inference.v1.lib.node_classification_inferencer
• gigl.src.inference.v1.lib.queries
• gigl.src.inference.v1.lib.transforms.batch_generator
• gigl.src.inference.v1.lib.transforms.utils
• gigl.src.inference.v1.lib.utils
• gigl.src.inference.v2.glt_inferencer
• gigl.src.mocking.dataset_asset_mocker
• gigl.src.mocking.dataset_asset_mocking_suite
• gigl.src.mocking.lib.constants
• gigl.src.mocking.lib.feature_handling
• gigl.src.mocking.lib.gbml_config
• gigl.src.mocking.lib.mock_input_for_data_preprocessor
• gigl.src.mocking.lib.mock_input_for_inference
• gigl.src.mocking.lib.mock_input_for_split_generator
• gigl.src.mocking.lib.mock_input_for_subgraph_sampler
• gigl.src.mocking.lib.mock_input_for_trainer
• gigl.src.mocking.lib.mock_output_for_inference
• gigl.src.mocking.lib.mocked_dataset_resources
• gigl.src.mocking.lib.pyg_datasets_forks
• gigl.src.mocking.lib.pyg_to_training_samples
• gigl.src.mocking.lib.tfrecord_io
• gigl.src.mocking.lib.user_defined_edge_sampling
• gigl.src.mocking.lib.versioning
• gigl.src.mocking.mocking_assets.mocked_datasets_for_pipeline_tests
• gigl.src.mocking.mocking_assets.passthrough_preprocessor_config_for_mocked_assets
• gigl.src.mocking.toy_asset_mocker
• gigl.src.post_process.lib.base_post_processor
• gigl.src.post_process.post_processor
• gigl.src.post_process.utils.component_runtime
• gigl.src.post_process.utils.cosine_similarity
• gigl.src.post_process.utils.unenumeration
• gigl.src.split_generator.split_generator
• gigl.src.subgraph_sampler.lib.ingestion_protocol
• gigl.src.subgraph_sampler.subgraph_sampler
• gigl.src.training.trainer
• gigl.src.training.v1.lib.base_trainer
• gigl.src.training.v1.lib.data_loaders.common
• gigl.src.training.v1.lib.data_loaders.node_anchor_based_link_prediction_data_loader
• gigl.src.training.v1.lib.data_loaders.rooted_node_neighborhood_data_loader
• gigl.src.training.v1.lib.data_loaders.supervised_node_classification_data_loader
• gigl.src.training.v1.lib.data_loaders.tf_records_iterable_dataset
• gigl.src.training.v1.lib.data_loaders.utils
• gigl.src.training.v1.lib.eval_metrics
• gigl.src.training.v1.lib.training_process
• gigl.src.training.v1.trainer
• gigl.src.training.v2.glt_trainer
• gigl.src.validation_check.config_validator
• gigl.src.validation_check.libs.frozen_config_path_checks
• gigl.src.validation_check.libs.name_checks
• gigl.src.validation_check.libs.resource_config_checks
• gigl.src.validation_check.libs.template_config_checks
• gigl.src.validation_check.libs.utils
• gigl.types.graph
• gigl.utils.data_splitters
• gigl.utils.share_memory
• snapchat.research.gbml.dataset_metadata_pb2
• snapchat.research.gbml.flattened_graph_metadata_pb2
• snapchat.research.gbml.gbml_config_pb2
• snapchat.research.gbml.gigl_resource_config_pb2
• snapchat.research.gbml.graph_schema_pb2
• snapchat.research.gbml.inference_metadata_pb2
• snapchat.research.gbml.postprocessed_metadata_pb2
• snapchat.research.gbml.preprocessed_metadata_pb2
• snapchat.research.gbml.subgraph_sampling_strategy_pb2
• snapchat.research.gbml.trained_model_metadata_pb2
• snapchat.research.gbml.training_samples_schema_pb2