jfrick commited on
Commit
9beeb42
·
verified ·
1 Parent(s): 6ad4fb0

python-124m-muon-5b-20261021-v1: 2750029614 training targets

Browse files
README.md CHANGED
@@ -11,7 +11,7 @@ tags:
11
  # Simple LLM Python 124M / 5B
12
 
13
  Run: `python-124m-muon-5b-20261021-v1`. This is an **intermediate** base-model checkpoint, freshly initialized.
14
- Processed **2,500,025,813 nonpadding training targets** in **42,519 optimizer updates**.
15
  The exact target is 5,000,000,000. This model is not instruction-tuned.
16
 
17
  124,389,632 parameters; 14 layers, width 768, 12 attention heads, RMSNorm,
 
11
  # Simple LLM Python 124M / 5B
12
 
13
  Run: `python-124m-muon-5b-20261021-v1`. This is an **intermediate** base-model checkpoint, freshly initialized.
14
+ Processed **2,750,029,614 nonpadding training targets** in **46,880 optimizer updates**.
15
  The exact target is 5,000,000,000. This model is not instruction-tuned.
16
 
17
  124,389,632 parameters; 14 layers, width 768, 12 attention heads, RMSNorm,
evaluation.json CHANGED
The diff for this file is too large to render. See raw diff
 
evaluations/002750029614-evaluated.json ADDED
The diff for this file is too large to render. See raw diff
 
metrics.jsonl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e8c69bacee5259d530ee64b94a5db49ccad3a89bc04f4d42ac7faf30f8e66e2e
3
- size 68693730
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ddcce92c81019a608eeba85cef0194d9c82fceb28ae2509cab63bbd68db5e4d0
3
+ size 75744653
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fe81807ad6958176c3836f364259ef9f1158beb2ed30a3da25c7d613a2337c71
3
  size 598238584
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:abcf556a9434e6fc77196a76841a237b65f3b974127b47b5130e6157fb09850a
3
  size 598238584
provenance.json CHANGED
@@ -47,22 +47,22 @@
47
  "tokenizer": "00e32adc13dab74f5a029c63304b4591bd48f7cd6b086b22d4624d244667b4da"
48
  },
49
  "torch_version": "2.6.0+cu124",
50
- "gpu": "NVIDIA H200",
51
  "parameters": 124389632,
52
- "resume_from": "/runs/python-124m-muon-5b-20261021-v1-s011/step-000038226.pt",
53
- "step": 42519,
54
- "tokens": 2500025813,
55
  "run_id": "python-124m-muon-5b-20261021-v1",
56
- "checkpoint": "/runs/python-124m-muon-5b-20261021-v1-s012/step-000042519.pt",
57
- "weights_sha256": "fe81807ad6958176c3836f364259ef9f1158beb2ed30a3da25c7d613a2337c71",
58
  "weight_roundtrip_exact": true,
59
  "cpu_logits_roundtrip_exact": true,
60
  "loss_token_exposures": {
61
- "python_implementation": 1625006416,
62
- "python_tests_examples": 375009387,
63
- "python_documentation": 250001615,
64
- "cs_math": 125003765,
65
- "general_education": 75000584,
66
- "books": 50004046
67
  }
68
  }
 
47
  "tokenizer": "00e32adc13dab74f5a029c63304b4591bd48f7cd6b086b22d4624d244667b4da"
48
  },
49
  "torch_version": "2.6.0+cu124",
50
+ "gpu": "NVIDIA H100 80GB HBM3",
51
  "parameters": 124389632,
52
+ "resume_from": "/runs/python-124m-muon-5b-20261021-v1-s012/step-000042519.pt",
53
+ "step": 46880,
54
+ "tokens": 2750029614,
55
  "run_id": "python-124m-muon-5b-20261021-v1",
56
+ "checkpoint": "/runs/python-124m-muon-5b-20261021-v1-s013/step-000046880.pt",
57
+ "weights_sha256": "abcf556a9434e6fc77196a76841a237b65f3b974127b47b5130e6157fb09850a",
58
  "weight_roundtrip_exact": true,
59
  "cpu_logits_roundtrip_exact": true,
60
  "loss_token_exposures": {
61
+ "python_implementation": 1787518112,
62
+ "python_tests_examples": 412503978,
63
+ "python_documentation": 275002110,
64
+ "cs_math": 137501054,
65
+ "general_education": 82501544,
66
+ "books": 55002816
67
  }
68
  }
runtime-cost-estimate.json CHANGED
@@ -1,11 +1,11 @@
1
  {
2
  "kind": "GPU compute estimate, not measured invoice",
3
- "training_worker_reported_seconds": 15882.104739504,
4
  "h100_usd_per_second": 0.001097,
5
  "excludes": [
6
  "GPU work outside recorded training segments",
7
  "CPU preparation and evaluation",
8
  "storage and transfer"
9
  ],
10
- "training_gpu_usd_estimate": 139.38135119388713
11
  }
 
1
  {
2
  "kind": "GPU compute estimate, not measured invoice",
3
+ "training_worker_reported_seconds": 17408.777889444,
4
  "h100_usd_per_second": 0.001097,
5
  "excludes": [
6
  "GPU work outside recorded training segments",
7
  "CPU preparation and evaluation",
8
  "storage and transfer"
9
  ],
10
+ "training_gpu_usd_estimate": 152.77943475776058
11
  }
training-curves.png CHANGED

Git LFS Details

  • SHA256: fb86aea03f2a59b602e2c11622da5256185961e3ef77ddf2fee33dab99074d7b
  • Pointer size: 131 Bytes
  • Size of remote file: 178 kB

Git LFS Details

  • SHA256: 81c2ea6e86222c7b22b728cd5e6120f970cb424b8cac7be1a06000c3fd75c701
  • Pointer size: 131 Bytes
  • Size of remote file: 177 kB
validation-curves.png CHANGED

Git LFS Details

  • SHA256: 124c6244ac39206b41a6f1540924b5961f32b353e1f14f78d325cdcc2e85ac5b
  • Pointer size: 131 Bytes
  • Size of remote file: 109 kB

Git LFS Details

  • SHA256: ff9d9153e0d2bf9c2099b73f36b02142a423b845c63fcc521cbe377a8da804e6
  • Pointer size: 131 Bytes
  • Size of remote file: 109 kB
validation.jsonl CHANGED
@@ -109,3 +109,13 @@
109
  {"step": 42000, "tokens": 2470760694, "target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "per_stratum": {"python_implementation": 1.6584197833332812, "python_tests_examples": 1.5135724393726473, "python_documentation": 2.6091162778057058, "cs_math": 2.7192630648603546, "general_education": 3.733178821266971, "books": 3.8689184870439757}, "stratum_target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "windows": 192, "per_source": {"python_implementation": 1.6584197833332812, "python_tests_examples": 1.5135724393726473, "python_documentation": 2.6091162778057058, "cs_math": 2.7192630648603546, "general_education": 3.733178821266971, "books": 3.8689184870439757}, "weighted_loss": 1.891257240475007}
110
  {"step": 42500, "tokens": 2499010434, "target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "per_stratum": {"python_implementation": 1.672793460030488, "python_tests_examples": 1.5085760987370946, "python_documentation": 2.6524121191564776, "cs_math": 2.6837753270545597, "general_education": 3.755705071581761, "books": 3.930422787980831}, "stratum_target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "windows": 192, "per_source": {"python_implementation": 1.672793460030488, "python_tests_examples": 1.5085760987370946, "python_documentation": 2.6524121191564776, "cs_math": 2.6837753270545597, "general_education": 3.755705071581761, "books": 3.930422787980831}, "weighted_loss": 1.9043117500058264}
111
  {"step": 42519, "tokens": 2500025813, "target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "per_stratum": {"python_implementation": 1.6665285634459845, "python_tests_examples": 1.5116764890828391, "python_documentation": 2.6236265730134023, "cs_math": 2.671596395204467, "general_education": 3.7442769424823514, "books": 3.91341999350956}, "stratum_target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "windows": 192, "per_source": {"python_implementation": 1.6665285634459845, "python_tests_examples": 1.5116764890828391, "python_documentation": 2.6236265730134023, "cs_math": 2.671596395204467, "general_education": 3.7442769424823514, "books": 3.91341999350956}, "weighted_loss": 1.896534224808541}
 
 
 
 
 
 
 
 
 
 
 
109
  {"step": 42000, "tokens": 2470760694, "target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "per_stratum": {"python_implementation": 1.6584197833332812, "python_tests_examples": 1.5135724393726473, "python_documentation": 2.6091162778057058, "cs_math": 2.7192630648603546, "general_education": 3.733178821266971, "books": 3.8689184870439757}, "stratum_target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "windows": 192, "per_source": {"python_implementation": 1.6584197833332812, "python_tests_examples": 1.5135724393726473, "python_documentation": 2.6091162778057058, "cs_math": 2.7192630648603546, "general_education": 3.733178821266971, "books": 3.8689184870439757}, "weighted_loss": 1.891257240475007}
110
  {"step": 42500, "tokens": 2499010434, "target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "per_stratum": {"python_implementation": 1.672793460030488, "python_tests_examples": 1.5085760987370946, "python_documentation": 2.6524121191564776, "cs_math": 2.6837753270545597, "general_education": 3.755705071581761, "books": 3.930422787980831}, "stratum_target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "windows": 192, "per_source": {"python_implementation": 1.672793460030488, "python_tests_examples": 1.5085760987370946, "python_documentation": 2.6524121191564776, "cs_math": 2.6837753270545597, "general_education": 3.755705071581761, "books": 3.930422787980831}, "weighted_loss": 1.9043117500058264}
111
  {"step": 42519, "tokens": 2500025813, "target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "per_stratum": {"python_implementation": 1.6665285634459845, "python_tests_examples": 1.5116764890828391, "python_documentation": 2.6236265730134023, "cs_math": 2.671596395204467, "general_education": 3.7442769424823514, "books": 3.91341999350956}, "stratum_target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "windows": 192, "per_source": {"python_implementation": 1.6665285634459845, "python_tests_examples": 1.5116764890828391, "python_documentation": 2.6236265730134023, "cs_math": 2.671596395204467, "general_education": 3.7442769424823514, "books": 3.91341999350956}, "weighted_loss": 1.896534224808541}
112
+ {"step": 42519, "tokens": 2500025813, "target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "per_stratum": {"python_implementation": 1.6665285634459845, "python_tests_examples": 1.5116764890828391, "python_documentation": 2.6236265730134023, "cs_math": 2.671596395204467, "general_education": 3.7442769424823514, "books": 3.91341999350956}, "stratum_target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "windows": 192, "per_source": {"python_implementation": 1.6665285634459845, "python_tests_examples": 1.5116764890828391, "python_documentation": 2.6236265730134023, "cs_math": 2.671596395204467, "general_education": 3.7442769424823514, "books": 3.91341999350956}, "weighted_loss": 1.896534224808541}
113
+ {"step": 43000, "tokens": 2527971083, "target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "per_stratum": {"python_implementation": 1.662284612737962, "python_tests_examples": 1.4964215858475767, "python_documentation": 2.600178602194451, "cs_math": 2.7010346768881215, "general_education": 3.724890235249983, "books": 3.9124972920963574}, "stratum_target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "windows": 192, "per_source": {"python_implementation": 1.662284612737962, "python_tests_examples": 1.4964215858475767, "python_documentation": 2.600178602194451, "cs_math": 2.7010346768881215, "general_education": 3.724890235249983, "books": 3.9124972920963574}, "weighted_loss": 1.8900144831200896}
114
+ {"step": 43500, "tokens": 2556315179, "target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "per_stratum": {"python_implementation": 1.6593065286970807, "python_tests_examples": 1.5066770522567425, "python_documentation": 2.600363345639107, "cs_math": 2.6659042873906227, "general_education": 3.7102325935298004, "books": 3.885079257506833}, "stratum_target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "windows": 192, "per_source": {"python_implementation": 1.6593065286970807, "python_tests_examples": 1.5066770522567425, "python_documentation": 2.600363345639107, "cs_math": 2.6659042873906227, "general_education": 3.7102325935298004, "books": 3.885079257506833}, "weighted_loss": 1.8868909133810865}
115
+ {"step": 44000, "tokens": 2585733432, "target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "per_stratum": {"python_implementation": 1.65876134067965, "python_tests_examples": 1.4965265943672301, "python_documentation": 2.590752870897446, "cs_math": 2.655538734468028, "general_education": 3.701792870486619, "books": 3.851070253224629}, "stratum_target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "windows": 192, "per_source": {"python_implementation": 1.65876134067965, "python_tests_examples": 1.4965265943672301, "python_documentation": 2.590752870897446, "cs_math": 2.655538734468028, "general_education": 3.701792870486619, "books": 3.851070253224629}, "weighted_loss": 1.8826012755890942}
116
+ {"step": 44500, "tokens": 2614214744, "target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "per_stratum": {"python_implementation": 1.6427071035977403, "python_tests_examples": 1.489512216835279, "python_documentation": 2.5802467922486985, "cs_math": 2.6597350916065308, "general_education": 3.6928744418986894, "books": 3.8242913897436672}, "stratum_target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "windows": 192, "per_source": {"python_implementation": 1.6427071035977403, "python_tests_examples": 1.489512216835279, "python_documentation": 2.5802467922486985, "cs_math": 2.6597350916065308, "general_education": 3.6928744418986894, "books": 3.8242913897436672}, "weighted_loss": 1.8694699447208536}
117
+ {"step": 45000, "tokens": 2643742535, "target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "per_stratum": {"python_implementation": 1.6504271266344603, "python_tests_examples": 1.489330347210624, "python_documentation": 2.5932837460476086, "cs_math": 2.6417932825022854, "general_education": 3.755251737032295, "books": 3.8289579849712263}, "stratum_target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "windows": 192, "per_source": {"python_implementation": 1.6504271266344603, "python_tests_examples": 1.489330347210624, "python_documentation": 2.5932837460476086, "cs_math": 2.6417932825022854, "general_education": 3.755251737032295, "books": 3.8289579849712263}, "weighted_loss": 1.8768319349342613}
118
+ {"step": 45500, "tokens": 2671594082, "target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "per_stratum": {"python_implementation": 1.6420125507057544, "python_tests_examples": 1.4934213985181783, "python_documentation": 2.575857575461065, "cs_math": 2.6424521159424, "general_education": 3.7008286913537, "books": 3.8692998818696074}, "stratum_target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "windows": 192, "per_source": {"python_implementation": 1.6420125507057544, "python_tests_examples": 1.4934213985181783, "python_documentation": 2.575857575461065, "cs_math": 2.6424521159424, "general_education": 3.7008286913537, "books": 3.8692998818696074}, "weighted_loss": 1.8694405894576969}
119
+ {"step": 46000, "tokens": 2699872835, "target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "per_stratum": {"python_implementation": 1.6516034398742783, "python_tests_examples": 1.484993995577513, "python_documentation": 2.572842087115055, "cs_math": 2.638648783390031, "general_education": 3.704827610137638, "books": 3.8900709382420082}, "stratum_target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "windows": 192, "per_source": {"python_implementation": 1.6516034398742783, "python_tests_examples": 1.484993995577513, "python_documentation": 2.572842087115055, "cs_math": 2.638648783390031, "general_education": 3.704827610137638, "books": 3.8900709382420082}, "weighted_loss": 1.8744542302048843}
120
+ {"step": 46500, "tokens": 2728888163, "target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "per_stratum": {"python_implementation": 1.6491192107410588, "python_tests_examples": 1.4691654664063583, "python_documentation": 2.5695742893452858, "cs_math": 2.664499644567642, "general_education": 3.692125639604816, "books": 3.8290243232799543}, "stratum_target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "windows": 192, "per_source": {"python_implementation": 1.6491192107410588, "python_tests_examples": 1.4691654664063583, "python_documentation": 2.5695742893452858, "cs_math": 2.664499644567642, "general_education": 3.692125639604816, "books": 3.8290243232799543}, "weighted_loss": 1.8698289737592964}
121
+ {"step": 46880, "tokens": 2750029614, "target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "per_stratum": {"python_implementation": 1.6467544658541027, "python_tests_examples": 1.4647498506032444, "python_documentation": 2.572555247388413, "cs_math": 2.648651639761941, "general_education": 3.69639633080658, "books": 3.806166974120848}, "stratum_target_counts": {"python_implementation": 55027, "python_tests_examples": 97392, "python_documentation": 28953, "cs_math": 50556, "general_education": 38282, "books": 256483}, "windows": 192, "per_source": {"python_implementation": 1.6467544658541027, "python_tests_examples": 1.4647498506032444, "python_documentation": 2.572555247388413, "cs_math": 2.648651639761941, "general_education": 3.69639633080658, "books": 3.806166974120848}, "weighted_loss": 1.866806216529206}