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
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆201Updated 4 months ago
- Butterfly matrix multiplication in PyTorch☆168Updated last year
- A Neural Operator-based Integrated Photonic Device Simulation Framework, NeurOLight NeurIPS 2022☆40Updated last year
- A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs☆162Updated 6 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆107Updated 2 weeks ago
- A header-only C++ library for sketching in randomized linear algebra☆84Updated last week
- Scientific Computational Imaging COde☆116Updated this week
- NVIDIA Math Libraries for the Python Ecosystem☆255Updated 2 weeks ago
- Functional models and algorithms for sparse signal processing☆89Updated last year
- Extending JAX with custom C++ and CUDA code☆388Updated 7 months ago
- Compositional Linear Algebra☆466Updated this week
- Collect optimizer related papers, data, repositories☆89Updated 4 months ago
- A high-performance C++ library for randomized numerical linear algebra☆67Updated this week
- Code for the book "The Elements of Differentiable Programming".☆77Updated last week
- Spyx: Spiking Neural Networks in JAX☆112Updated 5 months ago
- Python Algorithms for Randomized Linear Algebra☆53Updated last year
- Code for the paper "Analyzing inverse problems with invertible neural networks." (2018)☆94Updated 5 years ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆386Updated last month
- A parallel ODE solver for PyTorch☆251Updated 5 months ago
- High Granularity Quantizarion for Ultra-Fast Machine Learning Applications on FPGAs☆25Updated 2 months ago
- Matlab Algorithms for Randomized Linear Algebra☆15Updated last year
- Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024☆102Updated 4 months ago
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆138Updated 3 months ago
- A Library for Differentiable Logic Gate Networks☆656Updated last year
- A plugin for Jupyter Notebook to run CUDA C/C++ code☆217Updated 6 months ago
- ☆430Updated 3 months ago
- Kolmogorov-Arnold Networks built on JAX☆68Updated last week
- Pytorch RBF Layer implements a radial basis function layer in Pytorch. Radial Basis networks can be used to approximate functions.☆48Updated 3 years ago
- Improved LBFGS and LBFGS-B optimizers in PyTorch.☆60Updated 4 months ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆469Updated last week