rapidsai / wholegraph
WholeGraph - large scale Graph Neural Networks
☆99Updated this week
Related projects ⓘ
Alternatives and complementary repositories for wholegraph
- ☆72Updated 3 years ago
- A Python library transfers PyTorch tensors between CPU and NVMe☆96Updated this week
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆133Updated 2 months ago
- The official SALIENT system described in the paper "Accelerating Training and Inference of Graph Neural Networks with Fast Sampling and P…☆38Updated last year
- ☆101Updated 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…☆60Updated 2 years ago
- Artifact evaluation of the paper "Accelerating Training and Inference of Graph Neural Networks with Fast Sampling and Pipelining"☆24Updated 2 years ago
- Samples demonstrating how to use the Compute Sanitizer Tools and Public API☆65Updated last year
- 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
- Artifact for PPoPP20 "Understanding and Bridging the Gaps in Current GNN Performance Optimizations"☆39Updated 2 years ago
- distributed-embeddings is a library for building large embedding based models in Tensorflow 2.☆42Updated last year
- 🔮 Execution time predictions for deep neural network training iterations across different GPUs.☆54Updated last year
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆246Updated this week
- An experimental parallel training platform☆52Updated 7 months ago
- PyTorch distributed training acceleration framework☆32Updated this week
- A fast communication-overlapping library for tensor parallelism on GPUs.☆217Updated last week
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆146Updated this week
- Distributed Multi-GPU GNN Framework☆36Updated 4 years ago
- A GPU-accelerated graph learning library for PyTorch, facilitating the scaling of GNN training and inference.☆118Updated last week
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆75Updated this week
- Materials for learning SGLang☆75Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆268Updated this week
- Dynamic Memory Management for Serving LLMs without PagedAttention☆222Updated this week
- Chimera: Efficiently Training Large-Scale Neural Networks with Bidirectional Pipelines.☆44Updated 11 months ago
- Artifact for OSDI'23: MGG: Accelerating Graph Neural Networks with Fine-grained intra-kernel Communication-Computation Pipelining on Mult…☆37Updated 7 months ago
- PyTorch-Based Fast and Efficient Processing for Various Machine Learning Applications with Diverse Sparsity☆97Updated this week
- High performance RDMA-based distributed feature collection component for training GNN model on EXTREMELY large graph☆47Updated 2 years ago
- ☆79Updated 2 months ago
- Graph Sampling using GPU☆50Updated 2 years ago
- Set of datasets for the deep learning recommendation model (DLRM).☆40Updated last year