HazyResearch / mongoose
A Learnable LSH Framework for Efficient NN Training
☆30Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for mongoose
- Distributed ML Optimizer☆30Updated 3 years ago
- ☆15Updated 2 years ago
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆47Updated 2 years ago
- Code for the paper: https://arxiv.org/pdf/2309.06979.pdf☆16Updated 3 months ago
- Fast training of unitary deep network layers from low-rank updates☆28Updated last year
- "Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devices", official implementation☆28Updated last year
- Implementation of Hyena Hierarchy in JAX☆10Updated last year
- Code for Neural Execution Engines: Learning to Execute Subroutines☆16Updated 3 years ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆58Updated 3 years ago
- Parallelizing non-linear sequential models over the sequence length☆45Updated 3 weeks ago
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- Code Release for "Broken Neural Scaling Laws" (BNSL) paper☆57Updated last year
- LayerNorm(SmallInit(Embedding)) in a Transformer to improve convergence☆45Updated 2 years ago
- RWKV model implementation☆38Updated last year
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆43Updated last year
- Blog post☆16Updated 9 months ago
- ☆31Updated 10 months ago
- ☆29Updated 2 years ago
- ☆36Updated 10 months ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆79Updated 3 years ago
- Sparsity support for PyTorch☆31Updated this week
- ☆12Updated 2 years ago
- Locality sensitive hash functions for Tensorflow 2.0.☆13Updated 2 years ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆79Updated 9 months ago
- Make triton easier☆41Updated 5 months ago
- Official PyTorch code release for Implicit Gradient Transport, NeurIPS'19☆21Updated 5 years ago