rapidsai / wholegraph
WholeGraph - large scale Graph Neural Networks
☆101Updated 3 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
- distributed-embeddings is a library for building large embedding based models in Tensorflow 2.☆43Updated last year
- ☆107Updated 3 years ago
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆140Updated last week
- Artifact for OSDI'23: MGG: Accelerating Graph Neural Networks with Fine-grained intra-kernel Communication-Computation Pipelining on Mult…☆41Updated 11 months ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆313Updated this week
- A Python library transfers PyTorch tensors between CPU and NVMe☆107Updated 3 months ago
- Artifact for PPoPP20 "Understanding and Bridging the Gaps in Current GNN Performance Optimizations"☆39Updated 3 years ago
- oneCCL Bindings for Pytorch*☆89Updated last week
- The official SALIENT system described in the paper "Accelerating Training and Inference of Graph Neural Networks with Fast Sampling and P…☆38Updated last year
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆119Updated 2 years ago
- Synthesizer for optimal collective communication algorithms☆104Updated 11 months ago
- High performance RDMA-based distributed feature collection component for training GNN model on EXTREMELY large graph☆51Updated 2 years ago
- Training material for Nsight developer tools☆150Updated 7 months ago
- Graph Sampling using GPU☆51Updated 2 years ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆135Updated last year
- 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-Based Fast and Efficient Processing for Various Machine Learning Applications with Diverse Sparsity☆105Updated 3 weeks 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
- A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores☆50Updated last year
- A GPU-accelerated graph learning library for PyTorch, facilitating the scaling of GNN training and inference.☆127Updated last week
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆78Updated 3 months ago
- Dynamic Memory Management for Serving LLMs without PagedAttention☆308Updated 3 weeks ago
- ☆75Updated 2 years ago
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆201Updated last year
- GVProf: A Value Profiler for GPU-based Clusters☆49Updated 11 months ago
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆308Updated this week
- ☆141Updated last month
- A schedule language for large model training☆145Updated 8 months ago
- Samples demonstrating how to use the Compute Sanitizer Tools and Public API☆75Updated last year