jondurbin commited on
Commit
ea0d634
·
verified ·
1 Parent(s): e3b7208

Update model card and export table

Browse files
Files changed (1) hide show
  1. README.md +142 -0
README.md ADDED
@@ -0,0 +1,142 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ language:
4
+ - en
5
+ pipeline_tag: text-generation
6
+ tags:
7
+ - mixture-of-experts
8
+ - ternary
9
+ - decentralized-training
10
+ - base-model
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
+
27
+ Live training dashboard: [parallax.chutes.ai](https://parallax.chutes.ai).
28
+
29
+ ## Model
30
+
31
+ | | |
32
+ |---|---|
33
+ | Parameters | ~7.8B total, ~1.25B active per token |
34
+ | Layers | 64: 18 gated delta-rule (GDN2) recurrent, 8 sliding-window attention (window 2048), 6 sparse attention, 32 mixture-of-experts |
35
+ | Experts | 4096 routed experts (128 per MoE layer), 12 routed + 1 shared expert per token |
36
+ | Expert weights | ternary (-1, 0, +1) with per-row scales; at most two nonzero pairs in every group of eight |
37
+ | Width | d_model 1152; ReLU² expert activation |
38
+ | Tokenizer | Llama 3 tokenizer (vocabulary padded to 128,384); tied input/output embeddings |
39
+ | Training context | 4096 tokens |
40
+ | Output logit scale | learnable, capped at 3.0 |
41
+
42
+ ## Training
43
+
44
+ - Data: the same data mix as the earlier Parallax runs epsilon and kappa, not FineWeb-Edu. The main phase draws
45
+ from seven sources totalling about 1.09T tokens: two general pretraining mixtures of web, document, math and
46
+ code text (about 78% of the tokens), an additional web set, two math sets, a knowledge-focused set and a books
47
+ set. At about 899.6B tokens the data switches to two further sources for the rest of the run; the learning
48
+ rate does not change at that point. The run is planned as one pass of about 973B 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 (about 8.5M tokens per fleet step) for about the
55
+ first 100B tokens, then 20 (about 17M tokens per step) after a switch made while the run continued.
56
+ The fleet is 26 machines with 208 RTX 5090 GPUs.
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 once an hour while the run continues. Published exports are
63
+ never modified or removed.
64
+
65
+ | Export | Tokens | Step | Size | val_mix (nats/token) | 0-shot macro | 5-shot macro | Exported (UTC) |
66
+ |---|---:|---:|---:|---:|---:|---:|---|
67
+
68
+ - **val_mix**: mean cross-entropy (nats per token, lower is better) on a fixed held-out validation set
69
+ stratified over the nine sources of the data mix (2048 windows of 4096 tokens), the same set used for kappa.
70
+ It is not comparable with the FineWeb-Edu validation number published for theta.
71
+ - **0-shot macro**: mean over 11 tasks (ARC-Challenge, ARC-Easy, BoolQ, COPA, HellaSwag, LAMBADA,
72
+ OpenBookQA, PIQA, SciQ, SIQA, WinoGrande). **5-shot macro**: the same tasks without LAMBADA (10 tasks). Per task,
73
+ acc_norm is used for ARC, HellaSwag, OpenBookQA, PIQA and SciQ, and acc for the rest. Values are percentages.
74
+ - The scores come from the project's own scorer, which runs the native ternary experts. They track progress
75
+ within this run. Compare them with numbers from other evaluation harnesses with care.
76
+ - Benchmark prompts are scored with every run of two or more newlines collapsed to a single newline
77
+ (including the few-shot separator).
78
+ - A `-` means the export has not been scored yet. The table fills in as scores arrive.
79
+
80
+ ## Files and format
81
+
82
+ The files are in Parallax's **native compact export format** (inference only, no optimizer state). They
83
+ are byte-identical to the export the training system produced:
84
+
85
+ | File | Contents |
86
+ |---|---|
87
+ | `manifest.json` | export manifest: tensor inventory, per-file sha256 digests, token clock |
88
+ | `model_config.json` | model configuration |
89
+ | `coverage.json`, `layouts.json` | tensor coverage and expert frame layouts |
90
+ | `indexer.bundle` | sparse-attention indexer weights |
91
+ | `relay_pack/` | trunk (non-expert) weights in bf16, with their own manifest |
92
+ | `packed_experts.tar` | the 4096 routed experts (`packed_experts/*.t24p`, packed ternary codes and scales) |
93
+ | `SHA256SUMS` | sha256 of every file of the original export |
94
+ | `export_info.json` | step, tokens, time, sizes and digests of this export |
95
+
96
+ The only change from the original export is packaging. The 4096 expert files are stored in one
97
+ uncompressed tar to keep the repository's file count manageable. Extract it and check `SHA256SUMS`
98
+ as shown above. Every upload was checked against the training system's own digests before and
99
+ after it was published.
100
+
101
+ **Logit-scale bound.** The model's output logit scale is bounded: the forward pass uses
102
+ `exp(min(logit_scale_log, log 3.0))`. The stored trunk tensor is the raw training parameter (it can sit
103
+ slightly above the bound, e.g. from bf16 rounding), and each export records the bound in `manifest.json`
104
+ (`logit_scale`: `max`, `raw`, `effective`) and in `coverage.json` (`inference_policy.logit_scale_max`).
105
+ A loader must apply the recorded bound; the files themselves are left byte-identical.
106
+
107
+ ## Running it
108
+
109
+ This is a **base model only**. It is not chat- or instruction-tuned, so it does plain text
110
+ completion: give it the start of a text and it continues it. It will not follow instructions or hold a
111
+ conversation.
112
+
113
+ Standard `transformers` cannot load this format. Use our llama.cpp fork, [https://github.com/chutesai/llama.cpp](https://github.com/chutesai/llama.cpp), which adds a
114
+ dedicated runtime, `llama-parallax`, for CPU (x86-64, ARM64) and Apple GPUs (Metal):
115
+
116
+ ```bash
117
+ git clone https://github.com/chutesai/llama.cpp && cd llama.cpp
118
+ cmake -S . -B build -DCMAKE_BUILD_TYPE=Release -DGGML_METAL=ON # -DGGML_METAL=OFF for CPU only
119
+ cmake --build build --target llama-parallax -j 8
120
+ python tools/parallax/run.py --binary build/bin/llama-parallax --model parallax-t9.gguf \
121
+ --tokenizer tokenizer.json --backend metal --experts lut9 \
122
+ --prompt 'The capital of France is' --predict 64
123
+ ```
124
+
125
+ The runtime reads GGUF files converted from these exports. Ready-made GGUF files are not published
126
+ yet; `tools/parallax/README.md` in the fork covers conversion, options and the tokenizer.
127
+
128
+ ## Tech report
129
+
130
+ The Parallax tech report: [https://parallax.chutes.ai/tech-report.pdf](https://parallax.chutes.ai/tech-report.pdf). It is AI-generated from the project's
131
+ measurements and logs, and it is a living document that changes as the run progresses. It covers the predecessor run (kappa); lambda differs as described above.
132
+
133
+ ## Limitations
134
+
135
+ This is an early base model. It can produce incorrect, biased or nonsensical output. It has no
136
+ alignment or safety tuning, and it is small and far from converged.
137
+
138
+ ## License
139
+
140
+ MIT.
141
+
142
+ <sub>Table updated 2026-10-04 07:31 UTC.</sub>