rapidsai / wholegraph
WholeGraph - large scale Graph Neural Networks
☆101Updated 2 months ago
Alternatives and similar repositories for wholegraph:
Users that are interested in wholegraph are comparing it to the libraries listed below
- ☆73Updated 3 years ago
- ☆104Updated 3 years ago
- Graphiler is a compiler stack built on top of DGL and TorchScript which compiles GNNs defined using user-defined functions (UDFs) into ef…☆61Updated 2 years ago
- PyTorch-Direct code on top of PyTorch-1.8.0nightly (e152ca5) for Large Graph Convolutional Network Training with GPU-Oriented Data Commun…☆45Updated last year
- distributed-embeddings is a library for building large embedding based models in Tensorflow 2.☆43Updated last year
- Samples demonstrating how to use the Compute Sanitizer Tools and Public API☆75Updated last year
- oneCCL Bindings for Pytorch*☆88Updated last month
- The official SALIENT system described in the paper "Accelerating Training and Inference of Graph Neural Networks with Fast Sampling and P…☆38Updated last year
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆134Updated last year
- A Python library transfers PyTorch tensors between CPU and NVMe☆103Updated 2 months ago
- Artifact for PPoPP20 "Understanding and Bridging the Gaps in Current GNN Performance Optimizations"☆39Updated 3 years ago
- Graph Sampling using GPU☆51Updated 2 years ago
- Artifact for OSDI'23: MGG: Accelerating Graph Neural Networks with Fine-grained intra-kernel Communication-Computation Pipelining on Mult…☆38Updated 10 months ago
- PyTorch-Based Fast and Efficient Processing for Various Machine Learning Applications with Diverse Sparsity☆102Updated last week
- PArametrized Recommendation and Ai Model benchmark is a repository for development of numerous uBenchmarks as well as end to end nets for…☆128Updated this week
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆138Updated this week
- Memory Optimizations for Deep Learning (ICML 2023)☆62Updated 11 months ago
- A GPU-accelerated graph learning library for PyTorch, facilitating the scaling of GNN training and inference.☆125Updated 2 months ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆302Updated this week
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆294Updated this week
- Artifact evaluation of the paper "Accelerating Training and Inference of Graph Neural Networks with Fast Sampling and Pipelining"☆24Updated 2 years ago
- PyTorch Library for Low-Latency, High-Throughput Graph Learning on GPUs.☆296Updated last year
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆117Updated 2 years ago
- Distributed Multi-GPU GNN Framework☆36Updated 4 years ago
- Training material for Nsight developer tools☆146Updated 6 months ago
- Code for paper "Design Principles for Sparse Matrix Multiplication on the GPU" accepted to Euro-Par 2018☆72Updated 4 years ago
- Artifact for OSDI'21 GNNAdvisor: An Adaptive and Efficient Runtime System for GNN Acceleration on GPUs.☆64Updated last year
- ☆74Updated 2 years ago
- ☆50Updated 8 months ago
- ☆180Updated 7 months ago