jondurbin commited on
Commit
29057a0
·
verified ·
1 Parent(s): ad83d37

Update model card and export table

Browse files
Files changed (1) hide show
  1. README.md +16 -15
README.md CHANGED
@@ -11,16 +11,15 @@ tags:
11
  - research-checkpoint
12
  ---
13
 
14
- # Parallax 8B lambda: mid-training checkpoints
15
 
16
- These are research checkpoints from **lambda**, a training run of Parallax that is still in progress.
17
  Parallax is a mixture-of-experts language model trained on consumer GPUs spread over several
18
  countries. The hosts have no direct connections to each other and synchronize over the public internet.
19
 
20
  - **Base model.** It is pretrained on a mixture of web, document, math, code, knowledge-focused and book text. It is not instruction-tuned, chat-tuned or
21
  safety-tuned, and it will continue text rather than follow instructions.
22
- - **Mid-training.** Each export is a snapshot of a run that has not finished. Later exports are
23
- expected to differ, and quality between exports is not monotonic.
24
  - **Research artifact.** It is published so the training can be followed and inspected. It is
25
  not intended for production use.
26
 
@@ -45,28 +44,30 @@ Live training dashboard: [parallax.chutes.ai](https://parallax.chutes.ai).
45
 
46
  ## Training
47
 
48
- - Data: seven stationary sources totalling about 1.09T tokens: two general pretraining mixtures of web, document, math and code text, an additional web set, two math sets, a knowledge-focused set and a books set. At about 899.6B tokens the data switches to two further sources with no learning-rate event. Planned training totals about 972.9B tokens.
49
  - Optimization: the dense trunk is synchronized with a decoupled DiLoCo scheme over the internet. Routed
50
  experts are trained through low-rank adapters on the GPUs that use them, and the adapter updates are folded
51
  into full-precision masters that publish new ternary versions. The tech report describes the details.
52
  - Learning rate: linear warm-up from 0 to 6e-4 over the first 19.66B tokens, then 6e-4 until 400B tokens,
53
- 3e-4 until 700B and 1.5e-4 through the end of training. There is no final decay.
 
 
54
  - Batch: 10 gradient-accumulation micro-batches per step at launch (about 8.5M tokens per fleet step with 26 machines), switched to 20 at about 100B tokens on 2026-10-04 at 22:08 UTC. This gives about 17M tokens per step with all 26 machines; tokens per step vary with fleet size.
55
- - Fleet: 26 machines and 208 GPUs at launch. Three machines left on 2026-10-05 and their roles were reassigned automatically while training continued; the fleet has since comprised 23 machines and 184 GPUs.
56
  - Document masking: In training, attention and recurrent state are reset at document boundaries (up to 64 documents per 4096-token sequence; further boundaries are merged into the 64th document).
57
  - Output logits: the learnable output logit scale is capped at 3.0 in training and inference (see *Files and format*).
58
 
59
  ## Exports
60
 
61
  One folder per export under `exports/`, named by training tokens (rounded down to whole billions) and
62
- fleet step. A new export is added about every 30 minutes while the run continues. Published exports are
63
  never modified or removed.
64
 
65
  | Export | Tokens | Step | Size | Exported (UTC) |
66
  |---|---:|---:|---:|---|
67
- | [`901B-tokens_step73973_fc88abba`](https://huggingface.co/chutesai/parallax-8b-lambda/tree/main/exports/901B-tokens_step73973_fc88abba) **(latest)** | 901.5B | 73973 | 2.43 GB | 2026-10-10 01:44 |
68
  | [`901B-tokens_step73973`](https://huggingface.co/chutesai/parallax-8b-lambda/tree/main/exports/901B-tokens_step73973) | 901.5B | 73973 | 2.43 GB | 2026-10-10 01:17 |
69
- | [`900B-tokens_step73973`](https://huggingface.co/chutesai/parallax-8b-lambda/tree/main/exports/900B-tokens_step73973) | 900.8B | 73973 | 2.43 GB | 2026-10-10 00:45 |
70
  | [`898B-tokens_step73878`](https://huggingface.co/chutesai/parallax-8b-lambda/tree/main/exports/898B-tokens_step73878) | 898.7B | 73878 | 2.43 GB | 2026-10-10 00:18 |
71
  | [`895B-tokens_step73637`](https://huggingface.co/chutesai/parallax-8b-lambda/tree/main/exports/895B-tokens_step73637) | 896.0B | 73637 | 2.43 GB | 2026-10-09 23:49 |
72
  | [`892B-tokens_step73364`](https://huggingface.co/chutesai/parallax-8b-lambda/tree/main/exports/892B-tokens_step73364) | 892.8B | 73364 | 2.43 GB | 2026-10-09 23:18 |
@@ -341,13 +342,13 @@ never modified or removed.
341
 
342
  - This table lists the exports only. Validation loss and benchmark results for this run are shown on the live training dashboard.
343
 
344
- ## Latest export
345
 
346
- `exports/901B-tokens_step73973_fc88abba`: 901.5B training tokens, step 73973.
347
 
348
  ```bash
349
- hf download chutesai/parallax-8b-lambda --include "exports/901B-tokens_step73973_fc88abba/*" --local-dir parallax-8b-lambda
350
- cd parallax-8b-lambda/exports/901B-tokens_step73973_fc88abba
351
  tar -xf packed_experts.tar
352
  sha256sum -c --quiet SHA256SUMS # every file of the original export, byte for byte
353
  ```
@@ -512,4 +513,4 @@ alignment or safety tuning, and it is small and far from converged.
512
 
513
  MIT.
514
 
515
- <sub>Table updated 2026-10-10 01:52 UTC.</sub>
 
11
  - research-checkpoint
12
  ---
13
 
14
+ # Parallax 8B lambda: training checkpoints
15
 
16
+ These are research checkpoints from **lambda**, a training run of Parallax. Training was stopped at about 900B tokens, before the planned end of the run. The designated final checkpoint is step 73973 (900.8B tokens), marked in the table below. Exports saved later than the final checkpoint were saved after training stopped and are not the final checkpoint.
17
  Parallax is a mixture-of-experts language model trained on consumer GPUs spread over several
18
  countries. The hosts have no direct connections to each other and synchronize over the public internet.
19
 
20
  - **Base model.** It is pretrained on a mixture of web, document, math, code, knowledge-focused and book text. It is not instruction-tuned, chat-tuned or
21
  safety-tuned, and it will continue text rather than follow instructions.
22
+ - **Training snapshots.** Every export, the final one included, is a snapshot taken before the planned ~973B tokens. Quality between exports is not monotonic.
 
23
  - **Research artifact.** It is published so the training can be followed and inspected. It is
24
  not intended for production use.
25
 
 
44
 
45
  ## Training
46
 
47
+ - Data: seven stationary sources totalling about 1.09T tokens: two general pretraining mixtures of web, document, math and code text, an additional web set, two math sets, a knowledge-focused set and a books set. At about 899.6B tokens the data switched to two further sources while the learning-rate anneal that began at 877.28B continued. Training stopped at about 900B tokens, shortly after this switch. The planned training total was about 972.9B tokens.
48
  - Optimization: the dense trunk is synchronized with a decoupled DiLoCo scheme over the internet. Routed
49
  experts are trained through low-rank adapters on the GPUs that use them, and the adapter updates are folded
50
  into full-precision masters that publish new ternary versions. The tech report describes the details.
51
  - Learning rate: linear warm-up from 0 to 6e-4 over the first 19.66B tokens, then 6e-4 until 400B tokens,
52
+ 3e-4 until 700B and 1.5e-4 until 877.28B. A live anneal started at 877.28B tokens on 2026-10-09,
53
+ with multiplier m = 1 - 0.9 sqrt((T - 877.28B) / (972.90B - 877.28B)) and learning rate m times 1.5e-4.
54
+ When training stopped, the multiplier had reached 0.5489 (learning rate 8.234e-5). The anneal was partial.
55
  - Batch: 10 gradient-accumulation micro-batches per step at launch (about 8.5M tokens per fleet step with 26 machines), switched to 20 at about 100B tokens on 2026-10-04 at 22:08 UTC. This gives about 17M tokens per step with all 26 machines; tokens per step vary with fleet size.
56
+ - Fleet: 26 machines and 208 GPUs at launch. Seven machines were lost over 5-8 October, with roles reassigned automatically while training continued. At the end of the run, 19 machines were assigned, totalling 152 GPUs.
57
  - Document masking: In training, attention and recurrent state are reset at document boundaries (up to 64 documents per 4096-token sequence; further boundaries are merged into the 64th document).
58
  - Output logits: the learnable output logit scale is capped at 3.0 in training and inference (see *Files and format*).
59
 
60
  ## Exports
61
 
62
  One folder per export under `exports/`, named by training tokens (rounded down to whole billions) and
63
+ fleet step. Exports were added about every 30 minutes while training ran. Training has stopped; the final checkpoint is marked in the table. Published exports are
64
  never modified or removed.
65
 
66
  | Export | Tokens | Step | Size | Exported (UTC) |
67
  |---|---:|---:|---:|---|
68
+ | [`901B-tokens_step73973_fc88abba`](https://huggingface.co/chutesai/parallax-8b-lambda/tree/main/exports/901B-tokens_step73973_fc88abba) | 901.5B | 73973 | 2.43 GB | 2026-10-10 01:44 |
69
  | [`901B-tokens_step73973`](https://huggingface.co/chutesai/parallax-8b-lambda/tree/main/exports/901B-tokens_step73973) | 901.5B | 73973 | 2.43 GB | 2026-10-10 01:17 |
70
+ | [`900B-tokens_step73973`](https://huggingface.co/chutesai/parallax-8b-lambda/tree/main/exports/900B-tokens_step73973) **final** | 900.8B | 73973 | 2.43 GB | 2026-10-10 00:45 |
71
  | [`898B-tokens_step73878`](https://huggingface.co/chutesai/parallax-8b-lambda/tree/main/exports/898B-tokens_step73878) | 898.7B | 73878 | 2.43 GB | 2026-10-10 00:18 |
72
  | [`895B-tokens_step73637`](https://huggingface.co/chutesai/parallax-8b-lambda/tree/main/exports/895B-tokens_step73637) | 896.0B | 73637 | 2.43 GB | 2026-10-09 23:49 |
73
  | [`892B-tokens_step73364`](https://huggingface.co/chutesai/parallax-8b-lambda/tree/main/exports/892B-tokens_step73364) | 892.8B | 73364 | 2.43 GB | 2026-10-09 23:18 |
 
342
 
343
  - This table lists the exports only. Validation loss and benchmark results for this run are shown on the live training dashboard.
344
 
345
+ ## Final checkpoint
346
 
347
+ `exports/900B-tokens_step73973`: 900.8B training tokens, step 73973.
348
 
349
  ```bash
350
+ hf download chutesai/parallax-8b-lambda --include "exports/900B-tokens_step73973/*" --local-dir parallax-8b-lambda
351
+ cd parallax-8b-lambda/exports/900B-tokens_step73973
352
  tar -xf packed_experts.tar
353
  sha256sum -c --quiet SHA256SUMS # every file of the original export, byte for byte
354
  ```
 
513
 
514
  MIT.
515
 
516
+ <sub>Table updated 2026-10-10 02:05 UTC.</sub>