mberr / torch-max-memLinks
Decorators for maximizing memory utilization with PyTorch & CUDA
β17Updated this 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
Sorting:
- Taxonomy of Benchmarks in Graph Representation Learningβ19Updated 2 years ago
- ποΈ Hubness reduced nearest neighbor search for entity alignment with knowledge graph embeddingsβ25Updated last year
- Graph Positional and Structural Encoderβ51Updated 5 months ago
- π KG Inductive Link Prediction Challenge (ILPC) 2022β85Updated 3 years ago
- The SEMB library is an easy-to-use tool for getting and evaluating structural node embeddings in graphs.β18Updated 2 years ago
- π Lookup classes and instantiate them with styleβ64Updated 2 months ago
- β34Updated last month
- Multimodal Transformers for biomedical text and Knowledge Graph dataβ34Updated 2 years ago
- A set of benchmark repositories for node classification on knowledge graphs. To use, see https://github.com/pbloem/kgbench-loaderβ27Updated 2 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
- Geo2DR: A Python and PyTorch library for learning distributed representations of graphs.β46Updated last year
- Exploring the space of drug combinations to discover synergistic drugs using Active Learningβ24Updated 10 months ago
- Differentiable Euler Characteristic Transformβ17Updated last year
- Implementation of the Paper: "Parameterized Hypercomplex Graph Neural Networks for Graph Classification" by Tuan Le, Marco Bertolini, Fraβ¦β32Updated 3 years ago
- Using pretrained language models for biomedical knowledge graph completion.β47Updated 3 years ago
- β24Updated last year
- Deep Graph Mapper: Seeing Graphs through the Neural Lensβ58Updated last year
- Pytorch reimplementation of Molecule Attention Transformer, which uses a transformer to tackle the graph-like structure of moleculesβ59Updated 4 years ago
- Code for reproducing experiments in "On the Ability of Graph Neural Networks to Model Interactions Between Vertices"β25Updated last year
- Unofficially Implements https://arxiv.org/abs/2112.05682 to get Linear Memory Cost on Attention for PyTorchβ12Updated 3 years ago
- π Results for the Ranking Metrics submission @ GLB 2022β11Updated 3 years ago
- Code for "Molecular Hypergraph Grammar with Its Application to Molecular Optimization"β39Updated 6 months ago
- NeurIPS 2022: Tree Moverβs Distance: Bridging Graph Metrics and Stability of Graph Neural Networksβ37Updated last year
- 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
- Hierarchical Inter-Message Passing for Learning on Molecular Graphsβ79Updated 3 years ago
- TIP: Tri-graph Interaction Propagation model for Polypharmacy Side Eο¬ect Prediction (GRL@NeurIPS, 2019)β24Updated 2 years ago
- (Personalized) Page-Rank computation using PyTorchβ92Updated 2 years ago
- [ICML 2024] Recurrent Distance Filtering for Graph Representation Learningβ15Updated last year
- Implementation of Geometric Vector Perceptron, a simple circuit for 3d rotation equivariance for learning over large biomolecules, in Pytβ¦β74Updated 4 years ago
- Code for the paper "Query Embedding on Hyper-relational Knowledge Graphs"β33Updated last year