alisaaalehi / convolution_as_multiplication
Step by step explanation of 2D convolution implemented as matrix multiplication using toeplitz matrices
☆65Updated 2 years ago
Alternatives and similar repositories for convolution_as_multiplication
Users that are interested in convolution_as_multiplication are comparing it to the libraries listed below
Sorting:
- A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs☆167Updated 8 months ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆203Updated 5 months ago
- Python Algorithms for Randomized Linear Algebra☆54Updated 2 years ago
- General Matrix Multiplication using NVIDIA Tensor Cores☆17Updated 3 months ago
- Butterfly matrix multiplication in PyTorch☆170Updated last year
- Collect optimizer related papers, data, repositories☆91Updated 6 months ago
- TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆77Updated 11 months ago
- Scientific Computational Imaging COde☆128Updated 3 weeks ago
- Functional models and algorithms for sparse signal processing☆89Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆174Updated last week
- A Plug-and-play Lightweight tool for the Inference Optimization of Deep Neural networks☆41Updated 2 weeks ago
- A Neural Operator-based Integrated Photonic Device Simulation Framework, NeurOLight NeurIPS 2022☆40Updated last year
- A header-only C++ library for sketching in randomized linear algebra☆89Updated last month
- High Granularity Quantizarion for Ultra-Fast Machine Learning Applications on FPGAs☆27Updated last month
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆186Updated 5 months ago
- Implementation of "Gradients without backpropagation" paper (https://arxiv.org/abs/2202.08587) using functorch☆109Updated last year
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆52Updated 2 weeks ago
- Write a fast kernel and run it on Discord. See how you compare against the best!☆44Updated this week
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆262Updated this week
- High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial…☆44Updated 10 months ago
- Trying to find out what is the minimal model that can achieve 99% accuracy on MNIST dataset☆25Updated 6 years ago
- PyTorch Lightning implementation of the paper Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and H…☆29Updated 5 months ago
- Implementation of "NITI: Training Integer Neural Networks Using Integer-only Arithmetic" on arxiv☆81Updated 2 years ago
- Matlab Algorithms for Randomized Linear Algebra☆15Updated last year
- Improved LBFGS and LBFGS-B optimizers in PyTorch.☆63Updated 6 months ago
- Tensor-Train decomposition in pytorch☆62Updated 3 months ago
- A high-level, easy-to-deploy non-uniform Fast Fourier Transform in PyTorch.☆232Updated 5 months ago
- A plugin for Jupyter Notebook to run CUDA C/C++ code☆228Updated 8 months ago
- A Numpy implementation of a Convolutional Neural Network: slow & fast (im2col/col2im).☆58Updated last year
- Code for the paper "Analyzing inverse problems with invertible neural networks." (2018)☆95Updated 5 years ago