mberr / torch-max-memLinks
Decorators for maximizing memory utilization with PyTorch & CUDA
β17Updated 3 weeks 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β25Updated last year
- π KG Inductive Link Prediction Challenge (ILPC) 2022β85Updated 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
- (Personalized) Page-Rank computation using PyTorchβ92Updated 2 years ago
- Code for "R-GCN: The R Could Stand for Random"β37Updated 2 years ago
- NeurIPS 2022: Tree Moverβs Distance: Bridging Graph Metrics and Stability of Graph Neural Networksβ37Updated 2 years ago
- β26Updated last year
- π Lookup classes and instantiate them with styleβ67Updated this week
- π Results from the reproducibility and benchmarking studies presented in "Bringing Light Into the Dark: A Large-scale Evaluation of Knowβ¦β36Updated 3 years ago
- β29Updated 3 years ago
- Graph Transformers for Large Graphsβ22Updated last year
- Official Pytorch Implementation of GraphiTβ109Updated 4 years ago
- Multimodal Transformers for biomedical text and Knowledge Graph dataβ35Updated 2 years ago
- Graph Data Augmentation Library for PyTorch Geometricβ133Updated 3 years ago
- Y. Zhang, Q. Yao, J. Kwok. Bilinear Scoring Function Search for Knowledge Graph Learning. TPAMI 2022β69Updated last year
- Approximate subgraph motif mining through learnable edge contraction.β13Updated 2 years ago
- Using pretrained language models for biomedical knowledge graph completion.β47Updated 4 years ago
- Code for reproducing experiments in "On the Ability of Graph Neural Networks to Model Interactions Between Vertices"β25Updated 2 years ago
- Embedding graphs in symmetric spacesβ30Updated 4 years ago
- Graph Positional and Structural Encoderβ53Updated 9 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
- Official repository for the paper "Partition and Code: learning how to compress graphs" (NeurIPS'21) https://arxiv.org/abs/2107.01952β35Updated 3 years ago
- PyG re-implementation of Neural Bellman-Ford Networks (NeurIPS 2021)β75Updated 3 years ago
- Implementation of the Paper: "Parameterized Hypercomplex Graph Neural Networks for Graph Classification" by Tuan Le, Marco Bertolini, Fraβ¦β32Updated 4 years ago
- Exploring the space of drug combinations to discover synergistic drugs using Active Learningβ24Updated last year
- 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
- A Modular Framework for Learning on Multimodal Biomedical Knowledge Graphsβ18Updated last year
- How to Turn Your Knowledge Graph Embeddings into Generative Modelsβ53Updated last year
- Given an input graph (ArangoDB or PyG) it generates graph embeddings using Low-Code framework built on top of PyG.β68Updated last year