okarthikb / state-space-modelsLinks
☆27Updated last year
Alternatives and similar repositories for state-space-models
Users that are interested in state-space-models are comparing it to the libraries listed below
Sorting:
- Jax like function transformation engine but micro, microjax☆33Updated 10 months ago
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆105Updated 5 months ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆69Updated 4 months ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆19Updated last month
- gzip Predicts Data-dependent Scaling Laws☆35Updated last year
- H-Net Dynamic Hierarchical Architecture☆77Updated last month
- Collection of autoregressive model implementation☆86Updated 4 months ago
- ☆61Updated last year
- ☆82Updated last year
- DeMo: Decoupled Momentum Optimization