rapidsai / wholegraphLinks
WholeGraph - large scale Graph Neural Networks
☆104Updated 9 months ago
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
- Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…☆62Updated 2 months ago
- oneCCL Bindings for Pytorch*☆101Updated 3 weeks ago
- Set of datasets for the deep learning recommendation model (DLRM).☆47Updated 2 years ago
- Large Graph Convolutional Network Training with GPU-Oriented Data Communication Architecture (accepted by PVLDB)☆44Updated 2 years ago
- PyTorch Library for Low-Latency, High-Throughput Graph Learning on GPUs.☆301Updated 2 years ago
- Large scale graph learning on a single machine.☆164Updated 6 months ago
- PArametrized Recommendation and Ai Model benchmark is a repository for development of numerous uBenchmarks as well as end to end nets for…☆149Updated this week
- ☆109Updated 4 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 2 years ago
- distributed-embeddings is a library for building large embedding based models in Tensorflow 2.☆45Updated last year
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆159Updated 2 months 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
- Home for OctoML PyTorch Profiler☆114Updated 2 years ago
- A schedule language for large model training☆149Updated last week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆349Updated last week
- Python bindings for UCX☆138Updated this week
- A library of GPU kernels for sparse matrix operations.☆271Updated 4 years ago
- A Python library transfers PyTorch tensors between CPU and NVMe☆120Updated 9 months ago
- Microsoft Collective Communication Library☆67Updated 9 months ago
- Samples demonstrating how to use the Compute Sanitizer Tools and Public API☆86Updated last year
- An experimental parallel training platform☆54Updated last year
- A recommendation model kernel optimizing system☆10Updated 2 months ago
- ☆110Updated last year
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆404Updated this week
- Artifact for OSDI'23: MGG: Accelerating Graph Neural Networks with Fine-grained intra-kernel Communication-Computation Pipelining on Mult…☆40Updated last year
- Distributed preprocessing and data loading for language datasets☆39Updated last year
- ☆74Updated 5 months ago
- Artifact evaluation of the paper "Accelerating Training and Inference of Graph Neural Networks with Fast Sampling and Pipelining"☆23Updated 3 years ago
- 🏙 Interactive performance profiling and debugging tool for PyTorch neural networks.☆64Updated 7 months ago