catid / spectral_ssmLinks
Implementation of Spectral State Space Models
☆16Updated last year
Alternatives and similar repositories for spectral_ssm
Users that are interested in spectral_ssm are comparing it to the libraries listed below
Sorting:
- ☆28Updated last year
- GoldFinch and other hybrid transformer components☆46Updated last year
- ☆33Updated last year
- ☆32Updated last year
- Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"☆29Updated 5 years ago
- Triton Implementation of HyperAttention Algorithm☆48Updated last year
- A repository for research on medium sized language models.☆78Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆85Updated last year
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆38Updated last month
- Jax like function transformation engine but micro, microjax☆33Updated 9 months ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆101Updated 7 months ago
- ☆81Updated last year
- Official Code Repository for the paper "Key-value memory in the brain"☆27Updated 5 months ago
- HGRN2: Gated Linear RNNs with State Expansion☆55Updated 11 months ago
- ☆83Updated 11 months ago
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated last year
- ☆100Updated 2 weeks ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆19Updated 2 weeks ago
- Evaluating the Mamba architecture on the Othello game☆48Updated last year
- ☆37Updated last year
- My Implementation of Q-Sparse: All Large Language Models can be Fully Sparsely-Activated☆33Updated 11 months ago
- ☆69Updated 11 months ago
- A single repo with all scripts and utils to train / fine-tune the Mamba model with or without FIM☆56Updated last year
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated 6 months ago
- ☆53Updated 10 months ago
- Using FlexAttention to compute attention with different masking patterns☆44Updated 10 months ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆89Updated last year
- Utilities for Training Very Large Models☆58Updated 10 months ago
- This is the code that went into our practical dive using mamba as information extraction☆53Updated last year
- ☆31Updated last year