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
- A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs☆164Updated 7 months ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆204Updated 5 months ago
- A Neural Operator-based Integrated Photonic Device Simulation Framework, NeurOLight NeurIPS 2022☆40Updated last year
- Python Algorithms for Randomized Linear Algebra☆53Updated last year
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆107Updated last month
- Functional models and algorithms for sparse signal processing☆89Updated last year
- Distributed K-FAC preconditioner for PyTorch☆85Updated this week
- TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆77Updated 10 months ago
- Scientific Computational Imaging COde☆123Updated this week
- Newton and Quasi-Newton optimization with PyTorch☆342Updated last year
- High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial…☆44Updated 10 months ago
- ☆434Updated 4 months ago
- Butterfly matrix multiplication in PyTorch☆169Updated last year
- Code for the book "The Elements of Differentiable Programming".☆83Updated last month
- NVIDIA Math Libraries for the Python Ecosystem☆297Updated last month
- A simulation of the Kuramoto-Sivashinsky Equation in Python and MATLAB☆25Updated 6 years ago
- Implementation of "Gradients without backpropagation" paper (https://arxiv.org/abs/2202.08587) using functorch☆108Updated last year
- Matlab Algorithms for Randomized Linear Algebra☆15Updated last year
- Use scipy.optimize.minimize as a PyTorch Optimizer.☆72Updated 9 months ago
- Example to build PyTorch CUDA extension using CMake (with pybind11 and scikit-build)☆11Updated 4 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆125Updated 7 months ago
- Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024☆104Updated 5 months ago
- Extending JAX with custom C++ and CUDA code☆392Updated 8 months ago
- Machine Learning-Enabled Compact Photonic Tensor Core based on Programmable Multi-Operand Multimode Interference☆10Updated 7 months ago
- PyTorch implementation of Levenberg-Marquardt training algorithm☆57Updated 3 weeks ago
- A collection of papers in the area of photonic design automation☆60Updated last year
- Hundreds of strange attractors☆444Updated last week
- A Numpy implementation of a Convolutional Neural Network: slow & fast (im2col/col2im).☆56Updated last year
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆185Updated 4 months ago
- PyTorch library for solving imaging inverse problems using deep learning☆415Updated this week