okarthikb / state-space-models

Related projects ⓘ

Alternatives and complementary repositories for state-space-models

• joey00072 / ohara
  Collection of autoregressive model implementation
  ☆67Updated this week
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆26Updated 3 weeks ago
• xjdr-alt / llmri
  look how they massacred my boy
  ☆58Updated last month
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆84Updated last week
• xjdr-alt / simple_transformer
  Simple Transformer in Jax
  ☆119Updated 4 months ago
• euclaise / supertrainer2000
  ☆49Updated 8 months ago
• xjdr-alt / muzero_sketch
  ☆36Updated 3 months ago
• YuchenJin / llm.c
  LLM training in simple, raw C/CUDA
  ☆12Updated last month
• NVIDIA / ngpt
  Normalized Transformer (nGPT)
  ☆66Updated this week
• catid / lllm
  Latent Large Language Models
  ☆16Updated 2 months ago
• dvruette / barrel-rec-pytorch
  ☆53Updated 10 months ago
• mag- / gpu_benchmark
  Gpu benchmark
  ☆43Updated last month
• CarperAI / treasure_trove
  ☆22Updated last year
• NousResearch / StripedHyenaTrainer
  ☆57Updated 11 months ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆95Updated 6 months ago
• google-deepmind / mishax
  ☆101Updated 3 months ago
• google-deepmind / asyncdiloco
  ☆39Updated 10 months ago
• Zyphra / tree_attention
  Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters
  ☆104Updated last month
• andrew-silva / clean-rl-mlx
  Clean RL implementation using MLX
  ☆26Updated 8 months ago
• da-fr / arc-prize-2024
  Our solution for the arc challenge 2024
  ☆32Updated last week
• kyegomez / swarms-pytorch
  Swarming algorithms like PSO, Ant Colony, Sakana, and more in PyTorch 😊
  ☆111Updated last week
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆113Updated 7 months ago
• KellerJordan / Muon
  Muon optimizer for neural networks: >30% extra sample efficiency, <3% wallclock overhead
  ☆109Updated last week
• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆85Updated 2 months ago
• SonicCodes / lucid-v1
  realtime latent world model inference demo
  ☆34Updated last week
• Aleph-Alpha / scaling
  Scaling is a distributed training library and installable dependency designed to scale up neural networks, with a dedicated module for tr…
  ☆49Updated 3 weeks ago
• Aleph-Alpha / trigrams
  ☆43Updated 2 months ago
• SinatrasC / entropix-smollm
  smolLM with Entropix sampler on pytorch
  ☆139Updated 2 weeks ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆29Updated 2 weeks ago
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆70Updated 6 months ago