rapidsai / wholegraphLinks
WholeGraph - large scale Graph Neural Networks
☆106Updated last year
Alternatives and similar repositories for wholegraph
Users that are interested in wholegraph are comparing it to the libraries listed below
Sorting:
- ☆70Updated 4 years ago
- PyTorch Library for Low-Latency, High-Throughput Graph Learning on GPUs.☆302Updated 2 years ago
- distributed-embeddings is a library for building large embedding based models in Tensorflow 2.☆46Updated 2 years ago
- Large Graph Convolutional Network Training with GPU-Oriented Data Communication Architecture (accepted by PVLDB)☆44Updated 2 years ago
- ☆112Updated 4 years ago
- Large scale graph learning on a single machine.☆166Updated 9 months ago
- Set of datasets for the deep learning recommendation model (DLRM).☆48Updated 2 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…☆59Updated 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 2 years ago
- A Python library transfers PyTorch tensors between CPU and NVMe☆122Updated last year
- PArametrized Recommendation and Ai Model benchmark is a repository for development of numerous uBenchmarks as well as end to end nets for…☆154Updated this week
- oneCCL Bindings for Pytorch* (deprecated)☆103Updated last month
- Artifact evaluation of the paper "Accelerating Training and Inference of Graph Neural Networks with Fast Sampling and Pipelining"☆23Updated 3 years ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆365Updated this week
- A GPU-accelerated graph learning library for PyTorch, facilitating the scaling of GNN training and inference.☆146Updated 2 months ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆161Updated 2 months ago
- A schedule language for large model training☆151Updated 3 months ago
- A library of GPU kernels for sparse matrix operations.☆277Updated 5 years ago
- Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…☆63Updated 5 months ago
- Graph Sampling using GPU☆52Updated 3 years ago
- Artifact for OSDI'23: MGG: Accelerating Graph Neural Networks with Fine-grained intra-kernel Communication-Computation Pipelining on Mult…☆41Updated last year
- ☆113Updated last year
- Distributed Multi-GPU GNN Framework☆36Updated 5 years ago
- Chimera: bidirectional pipeline parallelism for efficiently training large-scale models.☆68Updated 8 months ago
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆182Updated last month
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆140Updated 2 years ago
- Microsoft Collective Communication Library☆66Updated last year
- Distributed preprocessing and data loading for language datasets☆39Updated last year
- NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process com…☆407Updated last month
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆121Updated this week