Exea LabsBlog
日本語Japanese version

Results

Every ExeaLM-versus-Llama run so far. Lower bits per byte is better.

ExeaLM
1.464
Llama
1.481

ExeaLM at 0.1B, same training time as a same-size Llama

01

Equal steps and equal compute

Equal compute (same training time)Equal steps
-0.04-0.020.00+0.02+0.04+0.06hybrid better ↑Llama better ↓13M27M48M90M
48M, first try (half attention)0.1B, hybrid at lr 1.5e-3
The hybrid's lead over Llama by model size. Above zero, the hybrid is better.
02

Every run

Bits per byte on held-out text, lower is better. Each row is one training run. Point at a column name to sort.

#RunModelLayoutBudgetlrSeed
1eqt_big_attn6TONNattention at 1, 2, 4, 5, 7, 8conv start2,000 s0.001501.3651.4351.3271.6551.446
2eqt_big_attn6_s1TONNattention at 1, 2, 4, 5, 7, 8conv start2,000 s0.001511.3671.4371.3281.6501.446
3eqt_big_llama_s1Llamaall attention2,000 s0.00111.3691.4391.3331.6651.451
4eqt_big_llamaLlamaall attention2,000 s0.00101.3701.4391.3321.6681.452
5eqt_100m_hybrid_lr1e3TONNattention at 1, 2, 4, 5, 7, 8, 10, 11, 13conv start3,000 s0.00101.3731.4501.3391.6921.464
6eqt_big_hybrid_lr15e4TONNattention at 1, 3, 5, 7, 9conv start2,000 s0.001501.3741.4531.3371.6991.466
7hconv_6kTONNattention at 1, 3, 5conv start6,000 steps0.00201.3901.4551.3431.6761.466
8eqt_big_hybridTONNattention at 1, 3, 5, 7, 9conv start2,000 s0.00201.3791.4561.3411.7121.472
9llama_6kLlamaall attention6,000 steps0.00201.3951.4591.3481.6861.472
10eqt_100m_hybridTONNattention at 1, 2, 4, 5, 7, 8, 10, 11, 13conv start3,000 s0.001501.3861.4621.3471.7021.474
11eqt_big_hybrid_lr1e3TONNattention at 1, 3, 5, 7, 9conv start2,000 s0.00101.3771.4581.3411.7231.475
12eqt_100m_llamaLlamaall attention3,000 s0.00101.3911.4681.3521.7131.481
13mid_hybridTONNattention at 1, 3, 5, 7conv start3,000 steps0.00211.4131.4811.3681.7021.491
14big_hybrid_3kTONNattention at 1, 3, 5, 7, 9conv start3,000 steps0.00201.3951.4771.3601.7351.492
15mid_hybridTONNattention at 1, 3, 5, 7conv start3,000 steps0.00201.4121.4841.3731.7021.493
16eqt_mid_hybridTONNattention at 1, 3, 5, 7conv start1,000 s0.00211.4151.4861.3731.7011.494
17eqt_mid_hybridTONNattention at 1, 3, 5, 7conv start1,000 s0.00201.4181.4921.3761.7191.502
18eqt_mid_llamaLlamaall attention1,000 s0.00111.4181.4901.3801.7201.502
19eqt_mid_llamaLlamaall attention1,000 s0.00101.4171.4941.3791.7301.505
20big_llama_3kLlamaall attention3,000 steps0.00101.4171.4961.3811.7271.505
21mid_llama_lr1e3Llamaall attention3,000 steps0.00111.4351.5081.3971.7411.520
22eqt_small_hybridTONNattention at 1, 3, 5conv start600 s0.00211.4451.5121.3951.7381.522
23eqt_small_llamaLlamaall attention600 s0.00201.4401.5111.3941.7451.522
24mid_hybrid_lr1e3TONNattention at 1, 3, 5, 7conv start3,000 steps0.00101.4261.5121.3901.7681.524
25hyb135TONNattention at 1, 3, 53,000 steps0.00201.4491.5161.3991.7361.525
26hyb135_searchTONN + searchattention at 1, 3, 53,000 steps0.00201.4481.5171.4001.7341.525
27mid_llama_lr1e3Llamaall attention3,000 steps0.00101.4341.5141.3981.7541.525
28hyb135_memTONN + memoryattention at 1, 3, 53,000 steps0.00201.4491.5181.4021.7341.526
29hconv_25TONNattention at 2, 5conv start3,000 steps0.00211.4521.5171.3991.7431.528
30hconv_attn4TONNattention at 1, 2, 4, 5conv start3,000 steps0.00211.4481.5161.3991.7491.528
31hconv_searchTONN + searchattention at 1, 3, 5conv start3,000 steps0.00211.4481.5161.4011.7511.529
32hconv_024TONNattention at 0, 2, 4conv start3,000 steps0.00201.4511.5221.4051.7411.530
33hyb135_convTONNattention at 1, 3, 5conv start3,000 steps0.00211.4511.5191.4021.7481.530
34eqt_small_hybridTONNattention at 1, 3, 5conv start600 s0.00201.4501.5241.4041.7461.531
35hconv_25TONNattention at 2, 5conv start3,000 steps0.00201.4511.5231.4061.7441.531
36hconv_searchTONN + searchattention at 1, 3, 5conv start3,000 steps0.00201.4521.5221.4061.7441.531
37hyb135_convTONNattention at 1, 3, 5conv start3,000 steps0.00201.4521.5231.4061.7451.532
38eqt_small_llamaLlamaall attention600 s0.00211.4491.5231.4011.7541.532
39hconv_attn4TONNattention at 1, 2, 4, 5conv start3,000 steps0.00201.4531.5271.4071.7601.537
40mid_llamaLlamaall attention3,000 steps0.00211.4471.5221.4011.7771.537
41mid_llamaLlamaall attention3,000 steps0.00201.4441.5241.4021.7781.537
42hconv_memTONN + memoryattention at 1, 3, 5conv start3,000 steps0.00201.4561.5321.4101.7641.540
43hconv_memTONN + memoryattention at 1, 3, 5conv start3,000 steps0.00211.4551.5281.4101.7731.542
44hconv_024TONNattention at 0, 2, 4conv start3,000 steps0.00211.4611.5301.4111.7781.545
45archLlamaall attention3,000 steps0.00201.4611.5341.4131.7751.546
46llama_lr3e3Llamaall attention3,000 steps0.00301.4651.5381.4151.7691.547
47llama_lr1e3Llamaall attention3,000 steps0.00101.4681.5481.4251.7761.554
48archLlamaall attention3,000 steps0.00211.4731.5451.4261.8031.562
49rp4TONNall TONN3,000 steps0.00201.4751.5551.4241.8691.581
50hyb135_searchTONN + searchattention at 1, 3, 53,000 steps0.00211.4551.5431.4071.9331.584
51archTONN + memoryall TONN3,000 steps0.00211.4751.5591.4211.9061.590
52hyb135_s1_curveTONNattention at 1, 3, 53,000 steps0.00211.4611.5511.4121.9491.593
53hyb135_s1TONNattention at 1, 3, 53,000 steps0.00211.4611.5511.4121.9501.594
54TONN + searchall TONN3,000 steps0.00201.4791.5661.4221.9481.604
55TONNall TONN3,000 steps0.00201.4831.5701.4241.9591.609
56TONN + searchall TONN3,000 steps0.00211.4741.5631.4171.9931.611
57TONNall TONN3,000 steps0.00211.4771.5671.4192.0001.616
58archTONN + memoryall TONN3,000 steps0.00201.4911.5781.4331.9641.616
03

Training curves

ExeaLM hybridLlama
4.05.06.07.08.005001000150020002500training time, secondsExeaLM hybrid 3.835Llama 3.963
Training loss against time. Switch to 48M for the first 2:1 win.
04

Speed

One layer, forward and backward (ms)
GI127 → 25 ms
SM145 → 45 ms
beforeafter the prefix scanattention block 18 ms
Training cost per token, relative to Llama (0.1B)
Pure TONN1.54× / 2.66×
Hybrid, half attention1.13× / 1.66×
Hybrid 2:1 (default)1.01× / 1.49×
memory circuits idleretrieval onLlama = 1
Circuit time before and after the prefix scan, and cost per token against Llama.
05

The sweep

03:0006:0009:0012:0015:0018:00Hybrid layers, random startConvolution-heavy startLearning rates, longer runsBigger models, equal stepsEqual computeLayoutsRetrieval and swarms on the hybridSpeed
Every run on the Mac, by start time.