jiachengpan / cudaMST
CUDA-accelerated minimum spanning tree algorithm -- data parallel Boruvka's algorithm
☆18Updated 8 years ago
Alternatives and similar repositories for cudaMST:
Users that are interested in cudaMST are comparing it to the libraries listed below
- EGGS, a method to speed up sparse matrix operations when the same sparsity is used for multiple times. This repo contains examples that s…☆25Updated 4 years ago
- Pytorch extension for Singular Value Decompostion (SVD) with LAPACK gesvd backend☆27Updated 4 years ago
- Computation using data flow graphs for scalable machine learning☆24Updated 6 years ago
- A PyTorch library for differentiable submodular minimization.☆58Updated 7 years ago
- [ICLR 2019] Learning Representations of Sets through Optimized Permutations☆36Updated 5 years ago
- Stochastic Deep Networks☆17Updated 5 years ago
- DDSL: Deep Differential Simplex Layer for Neural Networks☆55Updated 3 years ago
- Implementation of ConjugateGradients method using C and Nvidia CUDA☆49Updated 2 years ago
- Python Framework for sparse neural networks☆19Updated 7 years ago
- J. Solomon, F. de Goes, G. Peyré, M. Cuturi, A. Butscher, A. Nguyen, T. Du, L. Guibas. Convolutional Wasserstein Distances: Efficient Opt…☆103Updated 7 years ago
- ☆12Updated 5 years ago
- Keras implementation of Deep Wasserstein Embeddings☆47Updated 6 years ago
- Progressive Multigrid Eigensolver for Multiscale Angular Embedding Problems☆18Updated 8 years ago
- Kronecker Production Matrix for Approximate Nearest Neighbour Search☆12Updated 9 years ago
- Partially Adaptive Momentum Estimation method in the paper "Closing the Generalization Gap of Adaptive Gradient Methods in Training Deep …☆39Updated last year
- The Limited Multi-Label Projection Layer☆58Updated 6 months ago
- Implements an infinite sum of poisson-weighted convolutions☆26Updated 6 years ago
- A collection of adaptive sparse multi-scale solvers for optimal transport and related optimization problems.☆53Updated 3 years ago
- G. Peyré, L. Chizat, F-X. Vialard, J. Solomon, Quantum Optimal Transport for Tensor Field Processing, Arxiv, 2016☆10Updated 7 years ago
- Repository for our paper: Deep Graph Matching via Blackbox Differentiation of Combinatorial Solvers☆86Updated 4 years ago
- Easing non-convex optimization with neural networks.☆22Updated 6 years ago
- Legendre decomposition for tensors☆13Updated 6 years ago
- CUDA kernels for generalized matrix-multiplication in PyTorch☆79Updated 3 years ago
- A Unified, Systematic Framework of Structured Weight Pruning for DNNs☆22Updated 6 years ago
- ☆11Updated 3 years ago
- FluidNet re-written with ATen tensor lib☆51Updated 5 years ago
- PyTorch implementation of paper "Flat Metric Minimization with Applications in Generative Modeling"☆19Updated 5 years ago
- Codes for primal-dual method with linesearch☆8Updated 7 years ago
- A sparse KLU solver for PyTorch.☆62Updated 3 years ago
- A study of performance of optimal transport.☆10Updated 4 years ago