gigl.src.common.modeling_task_specs.utils.infer# Functions infer_root_embeddings infer_task_inputs infer_training_batch