mberr / torch-max-mem
Decorators for maximizing memory utilization with PyTorch & CUDA
β16Updated 2 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
- ποΈ 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
- Using pretrained language models for biomedical knowledge graph completion.β46Updated 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
- NeurIPS 2022: Tree Moverβs Distance: Bridging Graph Metrics and Stability of Graph Neural Networksβ36Updated 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
- Exploring the space of drug combinations to discover synergistic drugs using Active Learningβ24Updated 8 months ago
- Graph Positional and Structural Encoderβ50Updated 3 months ago
- Embedding graphs in symmetric spacesβ30Updated 3 years ago
- Improving Inductive Link Prediction Using Hyper-Relational Factsβ14Updated 2 years ago
- π Results for the Ranking Metrics submission @ GLB 2022β11Updated 3 years ago
- Graph Transformers for Large Graphsβ21Updated last year
- TIP: Tri-graph Interaction Propagation model for Polypharmacy Side Eο¬ect Prediction (GRL@NeurIPS, 2019)β24Updated 2 years ago
- PyG re-implementation of Neural Bellman-Ford Networks (NeurIPS 2021)β68Updated 2 years ago
- Locally corrected NystrΓΆm (LCN), as proposed in "Scalable Optimal Transport in High Dimensions for Graph Distances, Embedding Alignment, β¦β19Updated 2 years ago
- The repository implements the paper "Learning Graph Quantized Tokenizers for Transformers".β10Updated last month
- β24Updated last year
- β29Updated 2 years ago
- Multimodal Transformers for biomedical text and Knowledge Graph dataβ34Updated 2 years ago
- Geo2DR: A Python and PyTorch library for learning distributed representations of graphs.β46Updated last year
- Top1 Solution on OGB Challenge (Graph Property Prediction on HIV dataset)β10Updated 3 years ago
- (Personalized) Page-Rank computation using PyTorchβ90Updated 2 years ago
- π Easily pick a place to store data for your Python code.β36Updated last week
- Code for reproducing experiments in "On the Ability of Graph Neural Networks to Model Interactions Between Vertices"β25Updated last year
- π Lookup classes and instantiate them with styleβ64Updated 2 weeks ago
- A Modular Framework for Learning on Multimodal Biomedical Knowledge Graphsβ15Updated last year
- β18Updated 9 months ago
- Hierarchical Inter-Message Passing for Learning on Molecular Graphsβ79Updated 3 years ago
- The SEMB library is an easy-to-use tool for getting and evaluating structural node embeddings in graphs.β17Updated 2 years ago
- Replication attempt for the Protein Folding Model described in https://www.biorxiv.org/content/10.1101/2021.08.02.454840v1β37Updated 2 years ago