Update model card and export table
Browse files
README.md
CHANGED
|
@@ -11,16 +11,15 @@ tags:
|
|
| 11 |
- research-checkpoint
|
| 12 |
---
|
| 13 |
|
| 14 |
-
# Parallax 8B lambda:
|
| 15 |
|
| 16 |
-
These are research checkpoints from **lambda**, a training run of Parallax
|
| 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 |
-
- **
|
| 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
|
| 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
|
|
|
|
|
|
|
| 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.
|
| 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.
|
| 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)
|
| 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 |
-
##
|
| 345 |
|
| 346 |
-
`exports/
|
| 347 |
|
| 348 |
```bash
|
| 349 |
-
hf download chutesai/parallax-8b-lambda --include "exports/
|
| 350 |
-
cd parallax-8b-lambda/exports/
|
| 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
|
|
|
|
| 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>
|