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]
☆50Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for GraphFlow
- [ICLR 2019] Learning Representations of Sets through Optimized Permutations☆36Updated 5 years ago
- PyTorch-MPI-DDP-example☆17Updated 6 years ago
- CUDA templates for tile-sparse matrix multiplication based on CUTLASS.☆49Updated 6 years ago
- Experiments with Message Passing GNNs in C++ and PyTorch.☆23Updated 3 months ago
- Deriving Neural Architectures from Sequence and Graph Kernels☆60Updated 6 years ago
- kmeans clustering with multi-GPU capabilities☆116Updated last year
- Deep learning with a multiplication budget☆47Updated 6 years ago
- Structured matrices for compressing neural networks☆67Updated last year
- Discrete Object Generation with Reversible Inductive Construction (NeurIPS 2019)☆30Updated 3 years ago
- ☆66Updated last year
- Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch☆87Updated 3 years ago
- Notebooks for IPAM Tutorial, March 15 2019☆24Updated 5 years ago
- CUDA kernels for generalized matrix-multiplication in PyTorch☆79Updated 3 years ago
- Automatically insert nvtx ranges to PyTorch models☆17Updated 3 years ago
- Introduction to CUDA programming☆113Updated 7 years ago
- IPC: A Graph Data Set Compiled from International Planning Competitions☆44Updated 4 years ago
- Kernel Fusion and Runtime Compilation Based on NNVM☆69Updated 7 years ago
- Test winograd convolution written in TVM for CUDA and AMDGPU☆40Updated 6 years ago
- GPU Accelerated Subsampled Newton Method for Convex Optimization☆8Updated 6 years ago
- PyTorch Extension Library of Optimized Unique Operation☆37Updated 5 years ago
- ☆46Updated 5 years ago
- Efficient graph clustering software for normalized cut and ratio association on undirected graphs. Copyright(c) 2008 Brian Kulis, Yuqiang…☆22Updated 12 years ago
- implement distributed machine learning with Pytorch + OpenMPI☆51Updated 5 years ago
- Sample pytorch implementation of Covariant Compositional Networks☆13Updated 6 years ago
- Efficient Unitary Neural Network(EUNN) implementation in Tensorflow☆74Updated 5 years ago
- Simple example of implementing a new Tensorflow operation and its gradient in C++.☆56Updated 5 years ago
- 3D Roto-translation equivariant convolutions☆54Updated 5 years ago
- A Chainer extension for K-FAC☆20Updated 5 years ago
- Parallel network flows using OpenMP and CUDA.☆27Updated 5 years ago