PyTorch module that generates a subgraph originating from the UMLS Knowledge Graph (KG) when presented with a biomedical question (from MedQA). It utilizes Language Models (LMs) to solve the next relation prediction task and graph node embedding to infer the tail given a head and the predicted relation, essentially performing link prediction.