rapidsai / wholegraphLinks
WholeGraph - large scale Graph Neural Networks
☆104Updated 6 months ago
Alternatives and similar repositories for wholegraph
Users that are interested in wholegraph are comparing it to the libraries listed below
Sorting:
- ☆70Updated 3 years ago
- The official SALIENT system described in the paper "Accelerating Training and Inference of Graph Neural Networks with Fast Sampling and P…☆40Updated last year
- ☆106Updated 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
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆138Updated 2 years ago
- PyTorch-Based Fast and Efficient Processing for Various Machine Learning Applications with Diverse Sparsity☆108Updated last week
- Artifact for OSDI'23: MGG: Accelerating Graph Neural Networks with Fine-grained intra-kernel Communication-Computation Pipelining on Mult…☆40Updated last year
- Set of datasets for the deep learning recommendation model (DLRM).☆47Updated 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…☆44Updated last year
- oneCCL Bindings for Pytorch*☆97Updated last month
- Artifact for PPoPP20 "Understanding and Bridging the Gaps in Current GNN Performance Optimizations"☆38Updated 3 years ago
- A Python library transfers PyTorch tensors between CPU and NVMe☆115Updated 6 months ago
- Graph Sampling using GPU☆52Updated 3 years ago
- Ginex: SSD-enabled Billion-scale Graph Neural Network Training on a Single Machine via Provably Optimal In-memory Caching☆38Updated 10 months ago
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆121Updated 2 years ago
- A Data-Centric Compiler for Machine Learning☆83Updated last year
- Large scale graph learning on a single machine.☆163Updated 3 months ago
- PyTorch Library for Low-Latency, High-Throughput Graph Learning on GPUs.☆300Updated last year
- distributed-embeddings is a library for building large embedding based models in Tensorflow 2.☆44Updated last year
- PArametrized Recommendation and Ai Model benchmark is a repository for development of numerous uBenchmarks as well as end to end nets for…☆140Updated this week
- Samples demonstrating how to use the Compute Sanitizer Tools and Public API☆82Updated last year
- ☆79Updated 2 years ago
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆365Updated this week
- Artifact for OSDI'21 GNNAdvisor: An Adaptive and Efficient Runtime System for GNN Acceleration on GPUs.☆66Updated 2 years ago
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆146Updated last week
- A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores☆51Updated last year
- High performance RDMA-based distributed feature collection component for training GNN model on EXTREMELY large graph☆54Updated 2 years ago
- An Optimizing Compiler for Recommendation Model Inference☆23Updated last year
- A language and compiler for irregular tensor programs.☆137Updated 6 months ago
- Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…☆61Updated 2 months ago