Da1sypetals / cuda-Wavelet-KANLinks
CUDA implementation of Wavelet KAN.
☆15Updated last year
Alternatives and similar repositories for cuda-Wavelet-KAN
Users that are interested in cuda-Wavelet-KAN are comparing it to the libraries listed below
Sorting:
- Official Implementation for NorMuon paper☆39Updated 3 weeks ago
- ☆16Updated last year
- Oscillatory State-Space Models☆108Updated last month
- A State-Space Model with Rational Transfer Function Representation.☆83Updated last year
- Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)☆18Updated last year
- Implementation of xLSTM in Pytorch from the paper: "xLSTM: Extended Long Short-Term Memory"☆119Updated last month
- Kolmogorov-Arnold Networks (KAN) using orthogonal polynomials instead of B-splines.☆39Updated last year
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆15Updated 10 months ago
- Official Code Repository for the paper "Key-value memory in the brain"☆29Updated 9 months ago
- Official Implementation of Energy Transformer in PyTorch for Mask Image Reconstruction☆31Updated last year
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆59Updated 2 months ago
- Model hub for all your DiffeqML needs. Pretrained weights, modules, and basic inference infrastructure☆28Updated 2 years ago
- HGRN2: Gated Linear RNNs with State Expansion☆55Updated last year
- ☆23Updated last year
- Spectral Attention Autoregressive Model (SAAM)☆16Updated 3 years ago
- Official code for AL-PINNS: Augmented Lagrangian relaxation method for Physics-Informed Neural Networks☆12Updated 2 years ago
- ☆34Updated last year
- Inducing Point Operator Transformer: A Flexible and Scalable Architecture for Solving PDEs (AAAI 2024)☆15Updated last year
- Kolmogorov-Arnold Networks with various basis functions like B-Splines, Fourier, Chebyshev, Wavelets etc☆35Updated last year
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆66Updated last year
- A easy to use implementation of xLSTM☆62Updated 2 months ago
- FC-KAN: Function Combinations in Kolmogorov-Arnold Networks☆36Updated last week
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆127Updated 2 years ago
- ☆29Updated last year
- Solving High Frequency and Multi-Scale PDEs with Gaussian Processes (ICLR 2024)☆25Updated last year
- Kolmogorov-Arnold Networks (KAN) using Jacobi polynomials instead of B-splines.☆41Updated last year
- ☆97Updated last year
- Parallelizing non-linear sequential models over the sequence length☆55Updated 5 months ago
- This repository contains the codes to replicate the simulations from the paper: "Wav-KAN: Wavelet Kolmogorov-Arnold Networks". It showca…☆155Updated 5 months ago
- Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)☆81Updated last year