mberr / torch-max-memLinks
Decorators for maximizing memory utilization with PyTorch & CUDA
☆17Updated 3 months ago
Alternatives and similar repositories for torch-max-mem
Users that are interested in torch-max-mem are comparing it to the libraries listed below
Sorting:
- 🏘️ Hubness reduced nearest neighbor search for entity alignment with knowledge graph embeddings☆27Updated last year
- Multimodal Transformers for biomedical text and Knowledge Graph data☆34Updated 2 years ago
- 🏅 KG Inductive Link Prediction Challenge (ILPC) 2022☆85Updated 3 years ago
- (Personalized) Page-Rank computation using PyTorch☆93Updated 3 years ago
- Graph transport network (GTN), as proposed in "Scalable Optimal Transport in High Dimensions for Graph Distances, Embedding Alignment, an…☆15Updated 2 years ago
- Using pretrained language models for biomedical knowledge graph completion.☆47Updated 4 years ago
- Graph Data Augmentation Library for PyTorch Geometric☆133Updated 3 years ago
- Official Pytorch Implementation of GraphiT☆110Updated 4 years ago
- Graph Positional and Structural Encoder☆54Updated 11 months ago
- Official repository for the paper "Partition and Code: learning how to compress graphs" (NeurIPS'21) https://arxiv.org/abs/2107.01952☆35Updated 4 years ago
- Code for reproducing experiments in "On the Ability of Graph Neural Networks to Model Interactions Between Vertices"☆25Updated 2 years ago
- A Modular Framework for Learning on Multimodal Biomedical Knowledge Graphs☆18Updated 2 years ago
- Implementation of the Paper: "Parameterized Hypercomplex Graph Neural Networks for Graph Classification" by Tuan Le, Marco Bertolini, Fra…☆32Updated 4 years ago
- PyG re-implementation of Neural Bellman-Ford Networks (NeurIPS 2021)☆75Updated 3 years ago
- A set of benchmark repositories for node classification on knowledge graphs. To use, see https://github.com/pbloem/kgbench-loader☆28Updated 3 years ago
- Deep Graph Mapper: Seeing Graphs through the Neural Lens☆58Updated 2 years ago
- Improving Inductive Link Prediction Using Hyper-Relational Facts☆15Updated 3 years ago
- The official PyTorch implementation of "Pathfinder Discovery Networks for Neural Message Passing" (WebConf '21)☆59Updated 2 weeks ago
- NeurIPS 2022: Tree Mover’s Distance: Bridging Graph Metrics and Stability of Graph Neural Networks☆37Updated 2 years ago
- ☆26Updated last year
- 📊 Results from the reproducibility and benchmarking studies presented in "Bringing Light Into the Dark: A Large-scale Evaluation of Know…☆36Updated 3 years ago
- 📐 Results for the Ranking Metrics submission @ GLB 2022☆11Updated 3 years ago
- ☆34Updated 8 months ago
- Locally corrected Nyström (LCN), as proposed in "Scalable Optimal Transport in High Dimensions for Graph Distances, Embedding Alignment, …☆19Updated 2 years ago
- [ICML 2024] Recurrent Distance Filtering for Graph Representation Learning☆15Updated last year
- Embedding graphs in symmetric spaces☆30Updated 4 years ago
- Graph Transformers for Large Graphs☆22Updated last year
- Pytorch reimplementation of Molecule Attention Transformer, which uses a transformer to tackle the graph-like structure of molecules☆58Updated 5 years ago
- Y. Zhang, Q. Yao, J. Kwok. Bilinear Scoring Function Search for Knowledge Graph Learning. TPAMI 2022☆69Updated 2 years ago
- Implementation of Directional Graph Networks in PyTorch and DGL☆118Updated 4 years ago