Skip to content

Add large resumable MuonClip nanoGPT experiment - #126

Merged
charlesmartin14 merged 1 commit into
mainfrom
codex/large-muonclip-long-mps
Aug 30, 2026
Merged

Add large resumable MuonClip nanoGPT experiment#126
charlesmartin14 merged 1 commit into
mainfrom
codex/large-muonclip-long-mps

Conversation

@charlesmartin14

Copy link
Copy Markdown
Member

Adds a single-seed, MuonClip-only long-run experiment for Apple MPS.

Scale point:

  • 6 transformer blocks
  • 8 heads per block
  • width 384, context 512
  • 30,117,120 parameters
  • 512M pinned FineWeb-Edu training tokens
  • 62,500 updates at 8,192 tokens/update
  • 26 permanent WeightWatcher states covering all 36 transformer matrices

Also generalizes the shared model/config/WeightWatcher inventory from one block to multiple blocks, adds a full-size MPS smoke step, resumable detached-run/status tooling, and an atomically regenerated live HTML report.

Validation:

  • 31 targeted tests passed
  • exact scale arithmetic and 36-matrix inventory verified
  • live-report smoke test generated all seven plots

@charlesmartin14
charlesmartin14 merged commit 29c5e77 into main Aug 30, 2026
2 of 3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant