Update model card and export table
Browse files
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>
|