YiwenShaoStephen / NGD-SGDLinks
A Pytorch Implementation of Natural Gradient Descent
☆46Updated 6 years ago
Alternatives and similar repositories for NGD-SGD
Users that are interested in NGD-SGD are comparing it to the libraries listed below
Sorting:
- ☆164Updated 2 years ago
- Implementation of "Gradients without backpropagation" paper (https://arxiv.org/abs/2202.08587) using functorch☆114Updated 2 years ago
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆149Updated 2 years ago
- Online Normalization for Training Neural Networks (Companion Repository)☆86Updated 4 years ago
- A PyTorch Implementation of the Sparsemax operator (https://arxiv.org/pdf/1803.09820.pdf)☆34Updated 2 years ago
- Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)☆80Updated last year
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆60Updated 3 months ago
- Sequence Modeling with Structured State Spaces☆66Updated 3 years ago
- Distributed K-FAC preconditioner for PyTorch☆92Updated last week
- Applications using the GTN library and code to reproduce experiments in "Differentiable Weighted Finite-State Transducers"☆83Updated 3 years ago
- Fast Discounted Cumulative Sums in PyTorch☆96Updated 4 years ago
- Accelerated First Order Parallel Associative Scan☆192Updated last year
- A fully tensorized recurrent neural network using tensor-train decomposition☆26Updated 2 years ago
- Structured matrices for compressing neural networks☆67Updated 2 years ago
- Unofficial implementation of Linear Recurrent Units, by Deepmind, in Pytorch☆72Updated 7 months ago
- Automatic differentiation with weighted finite-state transducers.☆127Updated 3 years ago
- A minimal Pytorch Implementation of Stochastically Quantized Variational AutoEncoder (SQ-VAE) by Sony☆33Updated 2 years ago
- Implementation of Flow++ in PyTorch☆40Updated 6 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆132Updated 6 years ago
- Beyond Straight-Through☆105Updated 2 years ago
- Implementations of various linear RNN layers using pytorch and triton☆54Updated 2 years ago
- Adaptive Gradient Clipping☆153Updated 3 years ago
- Continuous Augmented Positional Embeddings (CAPE) implementation for PyTorch☆42Updated 2 years ago
- Rational Activation Functions - Replacing Padé Activation Units☆102Updated 8 months ago
- Jax/Flax implementation of Variational-DiffWave.☆40Updated 3 years ago
- Speech2Vec Reality Check☆84Updated 2 years ago
- PyTorch implementation of the paper "NanoFlow: Scalable Normalizing Flows with Sublinear Parameter Complexity." (NeurIPS 2020)☆66Updated 4 years ago
- A PyTorch implementation of "Continuous Relaxation Training of Discrete Latent Variable Image Models"☆75Updated 5 years ago
- Vector Quantized Autoregressive Predictive Coding (VQ-APC)☆37Updated 5 years ago
- custom cuda kernel for {2, 3}d relative attention with pytorch wrapper☆43Updated 5 years ago