A 49-million-parameter language model that implements key architectural ideas from Moonshot AI's Kimi K3 (including Kimi Delta Attention, Gated MLA, Stable LatentMoE, and per-head Muon) and can be trained from scratch on TinyStories using a single 8GB GPU.

Stars

46

7-day growth

No data

Forks

2

Open issues

0

License

No data

Last updated

2026-07-28

AI repository intelligence
FR-AI / ANALYSIS

Why it is worth attention

It makes complex frontier-model components—like linear attention with delta rule, sparse MoE with quantile balancing, and block attention residuals—accessible for hands-on study and modification on consumer hardware, without requiring massive compute.

Who it is for

  • Researchers and students exploring modern LLM architectures
  • Practitioners wanting to experiment with MoE and linear attention on limited budget
  • AI engineers interested in novel optimizers like Muon and Per-Head Muon
  • Hobbyists who want to train a non‑standard Transformer from scratch at home

Use cases

  • Understanding and iterating on Kimi K3-inspired components at small scale
  • Training a language model on a desktop GPU for educational purposes
  • Benchmarking training throughput and memory usage of hybrid KDA/MLA with MoE
  • Testing autoregressive decoding with recurrent attention caches

Strengths

  • Preserves the unusual hybrid attention and MoE design of a frontier model rather than replacing it with a standard Transformer
  • Runs on an 8GB GPU with verified throughput (901 tok/s at batch 16) and peak VRAM under 4 GB
  • Includes a live training dashboard and a full verification test suite (6 passing tests)
  • Well-documented with architecture diagrams, model card, and measured benchmarks

Considerations

  • Trained only on TinyStories, so generated text is limited to simple children's stories
  • Omitted the vision tower (401M parameters) and does not claim to reproduce Kimi K3's capabilities
  • First CUDA call requires ~75 seconds to compile Triton kernels, and the model is not optimized for production inference

README quick start

Quick start

Description

A 49M-parameter Kimi K3-inspired language model trainable on one 8GB GPU

Related repositories

Similar projects matched by category, topics, and programming language.

slvDev
Featured
slvDev GitHub avatar

esp32-ai

A 28.9 million parameter language model runs on an $8 ESP32-S3 microcontroller entirely on-device, generating simple stories at about 9.5 tokens per second.

AI & Machine LearningLarge Language Models
1,960
gavamedia
Featured
gavamedia GitHub avatar

deltafin

Deltafin is a research project that runs the 2.8-trillion-parameter Mixture-of-Experts model Kimi K3 on a single Apple Silicon Mac (e.g., M1 Max with 64 GB) at about 16 seconds per token, using exact, reproducible inference with local or streaming expert loading.

AI & Machine LearningLarge Language Models
304
jamesob
Featured
jamesob GitHub avatar

local-llm

A comprehensive guide for building and configuring a high-end local machine to run state-of-the-art LLMs, with detailed hardware choices, BIOS tuning, and Docker-based model serving.

AI & Machine LearningLarge Language Models
1,660