jfrick commited on
Commit
329ae19
·
verified ·
1 Parent(s): 1bb29b8

python-124m-muon-5b-20261021-v1: 2000044496 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 **1,750,010,347 nonpadding training targets** in **29,477 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,000,044,496 nonpadding training targets** in **33,868 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/002000044496-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:c0dd656f8d3d03c4148bdb72d54553a4f436c8d5fb2b40662bd246df1d621dc1
3
- size 47605051
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e6cd519d5606148c3d1b6ffe880a83dd48f4f625fef03fa5460df8d9656fe2ed
3
+ size 54705200
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0791d9601f6e02040b31d09b17080852e80eabe9a211da90eec1c612b61cd076
3
  size 598238584
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ee7e5b6464d223c25099ca4bcbb92272022cc8e39500f187ee6a436cf4d2c563
3
  size 598238584
provenance.json CHANGED
@@ -47,22 +47,22 @@
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-s008/step-000025134.pt",
53
- "step": 29477,
54
- "tokens": 1750010347,
55
  "run_id": "python-124m-muon-5b-20261021-v1",
56
- "checkpoint": "/runs/python-124m-muon-5b-20261021-v1-s009/step-000029477.pt",
57
- "weights_sha256": "0791d9601f6e02040b31d09b17080852e80eabe9a211da90eec1c612b61cd076",
58
  "weight_roundtrip_exact": true,
59
  "cpu_logits_roundtrip_exact": true,
60
  "loss_token_exposures": {
61
- "python_implementation": 1137504011,
62
- "python_tests_examples": 262504036,
63
- "python_documentation": 174999404,
64
- "cs_math": 87500573,
65
- "general_education": 52501748,
66
- "books": 35000575
67
  }
68
  }
 
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-s009/step-000029477.pt",
53
+ "step": 33868,
54
+ "tokens": 2000044496,
55
  "run_id": "python-124m-muon-5b-20261021-v1",
56
+ "checkpoint": "/runs/python-124m-muon-5b-20261021-v1-s010/step-000033868.pt",
57
+ "weights_sha256": "ee7e5b6464d223c25099ca4bcbb92272022cc8e39500f187ee6a436cf4d2c563",
58
  "weight_roundtrip_exact": true,
59
  "cpu_logits_roundtrip_exact": true,
60
  "loss_token_exposures": {
61
+ "python_implementation": 1300025113,
62
+ "python_tests_examples": 300009224,
63
+ "python_documentation": 200003637,
64
+ "cs_math": 100002345,
65
+ "general_education": 60002388,
66
+ "books": 40001789
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": 11565.736927455,
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": 101.50090727534509
11
  }
 
1
  {
2
  "kind": "GPU compute estimate, not measured invoice",
3
+ "training_worker_reported_seconds": 13010.55108537,
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": 114.18059632520713
11
  }
training-curves.png CHANGED

Git LFS Details

  • SHA256: f0a00475ab8bac2160be3bc16e6c6328119ae9afc1f72a2c64c619da2d159433
  • Pointer size: 131 Bytes
  • Size of remote file: 184 kB

Git LFS Details

  • SHA256: 5dc63abc8a8e305a98fd89b8618ffc407c9b054b9e009dece1c641fabbdb22bd
  • Pointer size: 131 Bytes
  • Size of remote file: 185 kB
validation-curves.png CHANGED

Git LFS Details

  • SHA256: 7a4ec0f86bf0dce397845279dc4aa7d558ab2d4bcac8ff9ba891b82a6305ea61
  • Pointer size: 131 Bytes
  • Size of remote file: 106 kB

Git LFS Details

  • SHA256: dd1e537d0329623c949651607e5c87d83b49b79a85443ec6e94d389c7745ab8e
  • Pointer size: 131 Bytes
  • Size of remote file: 108 kB
validation.jsonl CHANGED
@@ -76,3 +76,14 @@
76
  {"step": 28500, "tokens": 1692478231, "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.7574180564067199, "python_tests_examples": 1.5896200781995016, "python_documentation": 2.7066559490340487, "cs_math": 2.7939421301146874, "general_education": 3.846267246600883, "books": 3.988833050094938}, "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.7574180564067199, "python_tests_examples": 1.5896200781995016, "python_documentation": 2.7066559490340487, "cs_math": 2.7939421301146874, "general_education": 3.846267246600883, "books": 3.988833050094938}, "weighted_loss": 1.9862921282033577}
77
  {"step": 29000, "tokens": 1722016887, "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.7802874831173077, "python_tests_examples": 1.5980181085352476, "python_documentation": 2.729425466859814, "cs_math": 2.79540980915752, "general_education": 3.8600788159303407, "books": 4.008424649184839}, "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.7802874831173077, "python_tests_examples": 1.5980181085352476, "python_documentation": 2.729425466859814, "cs_math": 2.79540980915752, "general_education": 3.8600788159303407, "books": 4.008424649184839}, "weighted_loss": 2.0055734749120013}
78
  {"step": 29477, "tokens": 1750010347, "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.7467950020957397, "python_tests_examples": 1.5897608021121512, "python_documentation": 2.6845406729359866, "cs_math": 2.7847005012754265, "general_education": 3.850954334300287, "books": 4.067085262930438}, "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.7467950020957397, "python_tests_examples": 1.5897608021121512, "python_documentation": 2.6845406729359866, "cs_math": 2.7847005012754265, "general_education": 3.850954334300287, "books": 4.067085262930438}, "weighted_loss": 1.9784402993240409}
 
 
 
 
 
 
 
 
 
 
 
 
76
  {"step": 28500, "tokens": 1692478231, "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.7574180564067199, "python_tests_examples": 1.5896200781995016, "python_documentation": 2.7066559490340487, "cs_math": 2.7939421301146874, "general_education": 3.846267246600883, "books": 3.988833050094938}, "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.7574180564067199, "python_tests_examples": 1.5896200781995016, "python_documentation": 2.7066559490340487, "cs_math": 2.7939421301146874, "general_education": 3.846267246600883, "books": 3.988833050094938}, "weighted_loss": 1.9862921282033577}
77
  {"step": 29000, "tokens": 1722016887, "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.7802874831173077, "python_tests_examples": 1.5980181085352476, "python_documentation": 2.729425466859814, "cs_math": 2.79540980915752, "general_education": 3.8600788159303407, "books": 4.008424649184839}, "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.7802874831173077, "python_tests_examples": 1.5980181085352476, "python_documentation": 2.729425466859814, "cs_math": 2.79540980915752, "general_education": 3.8600788159303407, "books": 4.008424649184839}, "weighted_loss": 2.0055734749120013}
78
  {"step": 29477, "tokens": 1750010347, "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.7467950020957397, "python_tests_examples": 1.5897608021121512, "python_documentation": 2.6845406729359866, "cs_math": 2.7847005012754265, "general_education": 3.850954334300287, "books": 4.067085262930438}, "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.7467950020957397, "python_tests_examples": 1.5897608021121512, "python_documentation": 2.6845406729359866, "cs_math": 2.7847005012754265, "general_education": 3.850954334300287, "books": 4.067085262930438}, "weighted_loss": 1.9784402993240409}
79
+ {"step": 29477, "tokens": 1750010347, "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.7467950020957397, "python_tests_examples": 1.5897608021121512, "python_documentation": 2.6845406729359866, "cs_math": 2.7847005012754265, "general_education": 3.850954334300287, "books": 4.067085262930438}, "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.7467950020957397, "python_tests_examples": 1.5897608021121512, "python_documentation": 2.6845406729359866, "cs_math": 2.7847005012754265, "general_education": 3.850954334300287, "books": 4.067085262930438}, "weighted_loss": 1.9784402993240409}
80
+ {"step": 29500, "tokens": 1751283178, "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.7506204723265468, "python_tests_examples": 1.5822468261443003, "python_documentation": 2.695578254739497, "cs_math": 2.793008068636141, "general_education": 3.851033364958564, "books": 4.010159893028972}, "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.7506204723265468, "python_tests_examples": 1.5822468261443003, "python_documentation": 2.695578254739497, "cs_math": 2.793008068636141, "general_education": 3.851033364958564, "books": 4.010159893028972}, "weighted_loss": 1.9801827586489937}
81
+ {"step": 30000, "tokens": 1780115997, "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.7462845485028053, "python_tests_examples": 1.580393427558499, "python_documentation": 2.703077479726032, "cs_math": 2.807583857567328, "general_education": 3.8358382433349263, "books": 4.03439420281388}, "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.7462845485028053, "python_tests_examples": 1.580393427558499, "python_documentation": 2.703077479726032, "cs_math": 2.807583857567328, "general_education": 3.8358382433349263, "books": 4.03439420281388}, "weighted_loss": 1.9785939428678934}
82
+ {"step": 30500, "tokens": 1808072206, "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.7594301102195073, "python_tests_examples": 1.5712926581670874, "python_documentation": 2.7051107980869586, "cs_math": 2.7716356662738475, "general_education": 3.82137829544564, "books": 3.9732124173129897}, "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.7594301102195073, "python_tests_examples": 1.5712926581670874, "python_documentation": 2.7051107980869586, "cs_math": 2.7716356662738475, "general_education": 3.82137829544564, "books": 3.9732124173129897}, "weighted_loss": 1.9825219306997601}
83
+ {"step": 31000, "tokens": 1836670641, "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.7493851293380516, "python_tests_examples": 1.5767311094938015, "python_documentation": 2.69112109462696, "cs_math": 2.7533468950755875, "general_education": 3.8262904058410285, "books": 4.016261623881992}, "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.7493851293380516, "python_tests_examples": 1.5767311094938015, "python_documentation": 2.69112109462696, "cs_math": 2.7533468950755875, "general_education": 3.8262904058410285, "books": 4.016261623881992}, "weighted_loss": 1.9755033993631497}
84
+ {"step": 31500, "tokens": 1864848892, "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.744497996837419, "python_tests_examples": 1.5722828014781758, "python_documentation": 2.6849569591368985, "cs_math": 2.7510063092309998, "general_education": 3.8496743195184338, "books": 4.010365411188315}, "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.744497996837419, "python_tests_examples": 1.5722828014781758, "python_documentation": 2.6849569591368985, "cs_math": 2.7510063092309998, "general_education": 3.8496743195184338, "books": 4.010365411188315}, "weighted_loss": 1.9715096673506078}
85
+ {"step": 32000, "tokens": 1893563574, "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.7461878901870935, "python_tests_examples": 1.555592669002407, "python_documentation": 2.6764196073667614, "cs_math": 2.745716231694958, "general_education": 3.8375324225152934, "books": 3.9616053416856576}, "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.7461878901870935, "python_tests_examples": 1.555592669002407, "python_documentation": 2.6764196073667614, "cs_math": 2.745716231694958, "general_education": 3.8375324225152934, "books": 3.9616053416856576}, "weighted_loss": 1.9676468808025678}
86
+ {"step": 32500, "tokens": 1922324316, "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.7266952800541027, "python_tests_examples": 1.545292830337968, "python_documentation": 2.678006774398513, "cs_math": 2.81715600838983, "general_education": 3.821716686456245, "books": 3.9415612043792865}, "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.7266952800541027, "python_tests_examples": 1.545292830337968, "python_documentation": 2.678006774398513, "cs_math": 2.81715600838983, "general_education": 3.821716686456245, "books": 3.9415612043792865}, "weighted_loss": 1.9562870591264778}
87
+ {"step": 33000, "tokens": 1950659059, "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.759009503710785, "python_tests_examples": 1.5768835676583983, "python_documentation": 2.6973614729414646, "cs_math": 2.7902642995630775, "general_education": 3.836614855989556, "books": 3.9957197493656986}, "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.759009503710785, "python_tests_examples": 1.5768835676583983, "python_documentation": 2.6973614729414646, "cs_math": 2.7902642995630775, "general_education": 3.836614855989556, "books": 3.9957197493656986}, "weighted_loss": 1.984150915500071}
88
+ {"step": 33500, "tokens": 1978433425, "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.7402662197187007, "python_tests_examples": 1.5568571703293375, "python_documentation": 2.6798317536078136, "cs_math": 2.8720228350778405, "general_education": 3.801153109742646, "books": 3.9190878620001133}, "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.7402662197187007, "python_tests_examples": 1.5568571703293375, "python_documentation": 2.6798317536078136, "cs_math": 2.8720228350778405, "general_education": 3.801153109742646, "books": 3.9190878620001133}, "weighted_loss": 1.9687022860135113}
89
+ {"step": 33868, "tokens": 2000044496, "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.7381127430775816, "python_tests_examples": 1.5702197333558823, "python_documentation": 2.6799776620561504, "cs_math": 2.7968400148889274, "general_education": 3.8338019965578227, "books": 3.9896411339274924}, "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.7381127430775816, "python_tests_examples": 1.5702197333558823, "python_documentation": 2.6799776620561504, "cs_math": 2.7968400148889274, "general_education": 3.8338019965578227, "books": 3.9896411339274924}, "weighted_loss": 1.967952892529156}