mberr / torch-max-mem
Decorators for maximizing memory utilization with PyTorch & CUDA
β15Updated last week
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 9 months ago
- Graph Positional and Structural Encoderβ46Updated 3 weeks ago
- PyG re-implementation of Neural Bellman-Ford Networks (NeurIPS 2021)β65Updated 2 years ago
- π KG Inductive Link Prediction Challenge (ILPC) 2022β84Updated 2 years ago
- Using pretrained language models for biomedical knowledge graph completion.β46Updated 3 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
- Taxonomy of Benchmarks in Graph Representation Learningβ19Updated 2 years ago
- Improving Inductive Link Prediction Using Hyper-Relational Factsβ13Updated 2 years ago
- Code for the paper "Query Embedding on Hyper-relational Knowledge Graphs"β31Updated 9 months ago
- (Personalized) Page-Rank computation using PyTorchβ88Updated 2 years ago
- NeurIPS 2022: Tree Moverβs Distance: Bridging Graph Metrics and Stability of Graph Neural Networksβ36Updated last year
- Graph transport network (GTN), as proposed in "Scalable Optimal Transport in High Dimensions for Graph Distances, Embedding Alignment, anβ¦β16Updated last year
- Graph Transformers for Large Graphsβ20Updated 9 months ago
- [VLDB'23] SUREL+ is a novel set-based computation framework for scalable subgraph-based graph representation learning.β17Updated last year
- Embedding graphs in symmetric spacesβ30Updated 3 years ago
- Deep Graph Mapper: Seeing Graphs through the Neural Lensβ55Updated 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
- β29Updated 2 years ago
- Exploring the space of drug combinations to discover synergistic drugs using Active Learningβ24Updated 6 months ago
- Official implementation of NeurIPS'21: Implicit SVD for Graph Representation Learningβ20Updated 3 years ago
- DPPIN: A Biological Repository of Dynamic Protein-Protein Interaction Network Data, IEEE BigData 2022β27Updated 2 years ago
- DiffWire: Inductive Graph Rewiring via the LovΓ‘sz Bound. In Proceedings of the First Learning on Graphs Conference. 2022. Adrian Arnaiz-Rβ¦β18Updated 2 years ago
- Code for our paper "Attending to Graph Transformers"β85Updated last year
- Bi-Level Graph Neural Networks for Drug-Drug Interaction Prediction. ICML 2020 Graph Representation Learning and Beyond (GRL+) Workshopβ29Updated 4 years ago
- β24Updated 10 months ago
- β30Updated last year
- A library for subgraph GNN based on pygβ41Updated 2 months ago
- π Easily pick a place to store data for your Python code.β35Updated last week
- β23Updated 3 years ago
- β18Updated last year