HyTruongSon / GraphFlow
Deep Learning framework in C++/CUDA that supports symbolic/automatic differentiation, dynamic computation graphs, tensor/matrix operations accelerated by GPU and implementations of various state-of-the-art graph neural networks and other Machine Learning models including Covariant Compositional Networks For Learning Graphs [Risi et al]
☆53Updated 3 years ago
Alternatives and similar repositories for GraphFlow:
Users that are interested in GraphFlow are comparing it to the libraries listed below
- ☆23Updated 5 years ago
- Codebase associated with the PyTorch compiler tutorial☆46Updated 5 years ago
- Introduction to CUDA programming☆115Updated 7 years ago
- Parallel Tensor Infrastructure (ParTI!)☆28Updated 4 years ago
- Kernel Fusion and Runtime Compilation Based on NNVM☆70Updated 8 years ago
- Light-weight GPU kernel interface for graph operations☆15Updated 4 years ago
- MagmaDNN: a simple deep learning framework in c++☆49Updated 4 years ago
- HogWild++: A New Mechanism for Decentralized Asynchronous Stochastic Gradient Descent☆33Updated 8 years ago
- Sparse matrix-matrix multiplication on CPU+GPU systems.☆13Updated 11 years ago
- The pMMF Multiresolution Matrix Factorization Library☆26Updated 6 years ago
- kmeans clustering with multi-GPU capabilities☆120Updated last year
- G3: A Programmable GNN Training System on GPU☆43Updated 4 years ago
- Notebooks for IPAM Tutorial, March 15 2019☆24Updated 6 years ago
- Automatically insert nvtx ranges to PyTorch models☆17Updated 3 years ago
- Experiments with Message Passing GNNs in C++ and PyTorch.☆24Updated 8 months ago
- CUDA Matrix Factorization Library with Stochastic Gradient Descent (SGD)☆71Updated 7 years ago
- The Surprisingly ParalleL spArse Tensor Toolkit.☆70Updated 3 years ago
- IPC: A Graph Data Set Compiled from International Planning Competitions☆44Updated 5 years ago
- CUDA Sparse-Matrix Vector Multiplication, using Sliced Coordinate format☆21Updated 6 years ago
- ☆36Updated 10 months ago
- Deriving Neural Architectures from Sequence and Graph Kernels☆59Updated 7 years ago
- Distributed NMF/NTF Library☆44Updated 3 months ago
- ☆91Updated 8 years ago
- Enterprise: Breadth-First Graph Traversal on GPUs. SC'15.☆31Updated 7 years ago
- Sparse matrix computation library for GPU☆54Updated 4 years ago
- PyTorch-MPI-DDP-example☆17Updated 7 years ago
- A CUDA implementation of the PageRank Pipeline Benchmark☆32Updated 8 years ago
- CUDA templates for tile-sparse matrix multiplication based on CUTLASS.☆50Updated 7 years ago
- PyTorch Extension Library of Optimized Unique Operation☆37Updated 6 years ago
- ☆69Updated 2 years ago