mberr / torch-max-mem
Decorators for maximizing memory utilization with PyTorch & CUDA
β15Updated last month
Alternatives and similar repositories for torch-max-mem:
Users that are interested in torch-max-mem are comparing it to the libraries listed below
- ποΈ Hubness reduced nearest neighbor search for entity alignment with knowledge graph embeddingsβ25Updated 11 months ago
- Using pretrained language models for biomedical knowledge graph completion.β46Updated 3 years ago
- Exploring the space of drug combinations to discover synergistic drugs using Active Learningβ24Updated 7 months ago
- Graph Positional and Structural Encoderβ46Updated 2 months ago
- π KG Inductive Link Prediction Challenge (ILPC) 2022β84Updated 3 years ago
- Graph transport network (GTN), as proposed in "Scalable Optimal Transport in High Dimensions for Graph Distances, Embedding Alignment, anβ¦β16Updated last year
- A set of benchmark repositories for node classification on knowledge graphs. To use, see https://github.com/pbloem/kgbench-loaderβ25Updated 2 years ago
- Embedding graphs in symmetric spacesβ30Updated 3 years ago
- NeurIPS 2022: Tree Moverβs Distance: Bridging Graph Metrics and Stability of Graph Neural Networksβ36Updated last year
- Multimodal Transformers for biomedical text and Knowledge Graph dataβ34Updated 2 years ago
- π Results for the Ranking Metrics submission @ GLB 2022β11Updated 2 years ago
- π Results from the reproducibility and benchmarking studies presented in "Bringing Light Into the Dark: A Large-scale Evaluation of Knowβ¦β35Updated 2 years ago
- A Modular Framework for Learning on Multimodal Biomedical Knowledge Graphsβ13Updated last year
- Drug repurposing for COVID-19 using literature-based discoveryβ27Updated 2 years ago
- Code for reproducing experiments in "On the Ability of Graph Neural Networks to Model Interactions Between Vertices"β23Updated last year
- Improving Inductive Link Prediction Using Hyper-Relational Factsβ14Updated 2 years ago
- β13Updated 4 years ago
- Graph Transformers for Large Graphsβ21Updated 11 months ago
- β24Updated last year
- β29Updated 2 years ago
- PyG re-implementation of Neural Bellman-Ford Networks (NeurIPS 2021)β66Updated 2 years ago
- Code for the paper "Query Embedding on Hyper-relational Knowledge Graphs"β31Updated 10 months ago
- (Personalized) Page-Rank computation using PyTorchβ88Updated 2 years ago
- β18Updated last year
- TIP: Tri-graph Interaction Propagation model for Polypharmacy Side Eο¬ect Prediction (GRL@NeurIPS, 2019)β24Updated 2 years ago
- Biological Information Extraction from Large Language Models (LLMs)β10Updated last year
- β23Updated 3 years ago
- Deep Graph Mapper: Seeing Graphs through the Neural Lensβ57Updated last year
- π Lookup classes and instantiate them with styleβ64Updated 4 months ago
- Hierarchical Inter-Message Passing for Learning on Molecular Graphsβ77Updated 3 years ago