Alternatives and similar repositories for state-space-models

Users that are interested in state-space-models are comparing it to the libraries listed below

• JoeLi12345 / nGPT
  an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)
  ☆101Updated 3 months ago
• evanatyourservice / llm-jax
  Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.
  ☆17Updated 3 months ago
• joey00072 / ohara
  Collection of autoregressive model implementation
  ☆85Updated 2 months ago
• VatsaDev / NanoPoor
  NanoGPT-speedrunning for the poor T4 enjoyers
  ☆66Updated 2 months ago
• KhoomeiK / complexity-scaling
  gzip Predicts Data-dependent Scaling Laws
  ☆35Updated last year
• xjdr-alt / muzero_sketch
  ☆38Updated 10 months ago
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆32Updated 7 months ago
• tyler-romero / microR1
  Simple repository for training small reasoning models
  ☆33Updated 4 months ago
• NousResearch / StripedHyenaTrainer
  ☆61Updated last year
• dvruette / barrel-rec-pytorch
  ☆53Updated last year
• ClashLuke / SOAP
  ☆21Updated 7 months ago
• xjdr-alt / llmri
  look how they massacred my boy
  ☆63Updated 8 months ago
• kubernetes-bad / reward-composer
  Lego for GRPO
  ☆28Updated 3 weeks ago
• catid / lllm
  Latent Large Language Models
  ☆18Updated 10 months ago
• leloykun / modded-nanogpt
  NanoGPT (124M) quality in 2.67B tokens
  ☆28Updated last month
• euclaise / supertrainer2000
  ☆49Updated last year
• CarperAI / treasure_trove
  ☆22Updated last year
• Jaykef / Triton-nanoGPT
  Custom triton kernels for training Karpathy's nanoGPT.
  ☆19Updated 8 months ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated 7 months ago
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆82Updated 6 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆83Updated last year
• Pleias / Quest-Best-Tokens
  An introduction to LLM Sampling
  ☆78Updated 6 months ago
• Aleph-Alpha-Research / scaling
  Scaling is a distributed training library and installable dependency designed to scale up neural networks, with a dedicated module for tr…
  ☆63Updated 7 months ago
• minosvasilias / simple_grpo
  Simple GRPO scripts and configurations.
  ☆58Updated 4 months ago
• tokenbender / avataRL
  rl from zero pretrain, can it be done? we'll see.
  ☆56Updated this week
• jadechip / nanoXLSTM
  The simplest, fastest repository for training/finetuning medium-sized xLSTMs.
  ☆41Updated last year
• cloneofsimo / min-fsdp
  ☆78Updated 11 months ago
• notarussianteenager / srf-attention
  Simplex Random Feature attention, in PyTorch
  ☆74Updated last year
• xjdr-alt / simple_transformer
  Simple Transformer in Jax
  ☆137Updated last year
• ethansmith2000 / TransformerExperiments
  ☆19Updated last month