alisaaalehi / convolution_as_multiplicationLinks
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:
- Matlab Algorithms for Randomized Linear Algebra☆15Updated last year
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆204Updated 7 months ago
- Python Algorithms for Randomized Linear Algebra☆54Updated 2 years ago
- A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs☆171Updated 9 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆178Updated 2 weeks ago
- Butterfly matrix multiplication in PyTorch☆169Updated last year
- A header-only C++ library for sketching in randomized linear algebra☆91Updated 3 months ago
- TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆79Updated last year
- A Neural Operator-based Integrated Photonic Device Simulation Framework, NeurOLight NeurIPS 2022☆40Updated last year
- Learning Activation Functions in Deep (Spline) Neural Networks☆28Updated 2 years ago
- A plugin for Jupyter Notebook to run CUDA C/C++ code☆233Updated 9 months ago
- General Matrix Multiplication using NVIDIA Tensor Cores☆18Updated 5 months ago
- High Granularity Quantizarion for Ultra-Fast Machine Learning Applications on FPGAs☆30Updated 2 months ago
- A high-performance C++ library for randomized numerical linear algebra☆72Updated 2 weeks ago
- Distributed K-FAC preconditioner for PyTorch☆87Updated this week
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆126Updated 2 weeks ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆111Updated 3 months ago
- ☆436Updated 8 months ago
- Functional models and algorithms for sparse signal processing☆90Updated last year
- GPU/TPU accelerated nonlinear least-squares curve fitting using JAX☆57Updated 2 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆128Updated 9 months ago
- Code for the book "The Elements of Differentiable Programming".☆88Updated last week
- ☆193Updated this week
- This page contains Python codes for the chapters appearing in all 3 volumes of the work "Sayed, Ali. H., Inference and Learning from Data…☆24Updated last year
- Spyx: Spiking Neural Networks in JAX☆122Updated 8 months ago
- Compositional Linear Algebra☆477Updated last month
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆188Updated 6 months ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆481Updated last week
- Extending JAX with custom C++ and CUDA code☆396Updated 10 months ago
- A projection-based framework for gradient-free and parallel learning☆68Updated last week