AETHORIA-AI/TR-HASH-MoE-200M-160B-SFT

Model

5

stars

40

commits

3

linked in READMEs

Aug 28, 2026

updated

conversational
custom-code
custom_code
full-parameter-finetuning
gqa
mixture-of-experts
pytorch
safetensors
supervised-finetuning
text-generation
transformers
tr-hash
tr_hash_moe

README

TR-HASH MoE 200M — 160B-source Full SFT (32K release)

Paper

Deterministic multi-hash routing supports long-horizon training in a compact language model

Preprint (DOI) · Model weights · Code

Full-parameter instruction SFT of AETHORIA-AI/TR-HASH-MoE-200M-160B-Refinement on the audited AETHORIA-AI/TR-HASH-MoE-200M-SFT-v2-300K. This release is not LoRA or QLoRA: all 201.2M model parameters were trainable.

The root model.safetensors is the epoch 3 / step 5,982 checkpoint, selected for the strongest full-PIQA result and lowest held-out SFT loss across the three epochs.

Tokenizer: use AETHORIA-AI/TR-HASH-Tokenizer-32K with this 32,000-token model.

Results

EpochStepHeld-out SFT lossSFT pplPIQA accPIQA acc_norm
11,9940.9909432.6967.90%68.93%
23,9880.9639122.6267.85%68.82%
35,9820.9596172.6168.01%69.10%

Released checkpoint benchmark panel

BenchmarkSplitExamplesAccuracyAccuracy (length-normalized)Evaluation backend
PIQAvalidation1,83868.01%69.10%PyTorch FP16, custom Triton
ARC-Easytest2,37657.24%52.86%MLX FP16
ARC-Challengetest1,17227.13%29.01%MLX FP16
ARC Combined (micro)test3,54847.29%44.98%Derived from both ARC test splits
HellaSwagvalidation10,04233.21%38.74%MLX FP16

All benchmark evaluations use zero-shot causal continuation log-likelihood, no chat template and a maximum sequence length of 2,048. Accuracy selects the choice with the highest total continuation log-likelihood; the normalized metric selects by mean continuation log-likelihood per scored token. PIQA was evaluated from the native epoch-3 checkpoint. ARC and HellaSwag were evaluated from an FP16 MLX conversion of the same root F32 SafeTensors weights. ARC Combined is the micro-average over all 3,548 ARC-Easy and ARC-Challenge test examples, not the arithmetic mean of the two percentages. Machine-readable reports are published under reports/sft-v2-300k/.

Compact-model comparison

Combined ARC comparison from 124M to 774M parameters

ModelParametersPIQA accARC-Easy accARC-Challenge accCombined ARC accHellaSwag acc
TR-HASH MoE 200M Full SFT201.2M68.01%57.24%27.13%47.29%33.21%
GPT-2 Large774M53.11%21.76%42.76%
Pythia-410M410M52.02%21.42%41.91%
GPT-2 Medium355M49.16%21.67%40.08%
OPT-350M350M43.98%20.82%36.33%
GPT-2 Small124M62.89%43.81%19.03%≈35.63%28.92%
OPT-125M125M63.00%43.60%19.10%≈35.51%29.20%
Pythia-160M160M62.73%43.52%18.77%35.34%

Combined ARC is weighted by the public test-set sizes (2,376 ARC-Easy and 1,172 ARC-Challenge examples). GPT-2 Small and OPT reference scores come from the AMD-LLM lm-evaluation-harness comparison; their combined values are approximate because the published component scores are rounded. Pythia-160M uses EleutherAI's official zero-shot result. GPT-2 Medium, GPT-2 Large and Pythia-410M were evaluated locally in MLX FP16 with the same causal-choice evaluator as TR-HASH. OPT-350M used the identical prompt and scoring formula in PyTorch MPS FP16 because MLX does not implement the OPT architecture. The 124M–160M references were reported through lm-evaluation-harness, so that part of the comparison is informative rather than a claim of bit-identical evaluation runtimes.

Training recipe

SettingValue
MethodFull-parameter supervised fine-tuning
SourceRefinement step 8,156 (about 162B prior token exposures)
Dataset300,000 train / 3,000 held-out examples
Tokenized corpus202,948,693 train tokens; no truncation
SupervisionFinal assistant turn only; prior assistant turns masked
Epochs3
Context2,048 tokens
TokenizerTR-HASH-Tokenizer-32K; EOS </s> (ID 0)
OptimizerAdamW, betas 0.9 / 0.95, weight decay 0.1
LR2e-5 peak, 3% warmup, continuous cosine decay
PrecisionBF16 training
Root SafeTensors precisionfloat32
KernelsLiger required; custom Triton enabled

Architecture and loading

201.2M parameters, 16 decoder layers, GQA (14 query heads / 2 KV heads), four stored deterministic token-ID-routed experts with top-2 activation, an always-on shared SwiGLU path and tied embeddings. The persisted multi-hash routing tables are part of the checkpoint.

The repository includes an autonomous Transformers adapter. Load it with:

from transformers import AutoModelForCausalLM, AutoTokenizer

repo = "AETHORIA-AI/TR-HASH-MoE-200M-160B-SFT"
tokenizer = AutoTokenizer.from_pretrained(repo, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(repo, trust_remote_code=True)

License

The released model weights and repository code are licensed under CC BY-NC 4.0 as reproduced in LICENSE. Commercial use is not permitted under this license. Source datasets retain their own licenses and terms; see the dataset manifest for the per-source audit.

Citation

@article{Peyriguere_2026,
  title     = {Deterministic multi-hash routing supports long-horizon training in a compact language model},
  author    = {Peyriguere, Boris},
  year      = {2026},
  month     = aug,
  publisher = {Springer Science and Business Media LLC},
  doi       = {10.21203/rs.3.rs-10788774/v1},
  url       = {https://doi.org/10.21203/rs.3.rs-10788774/v1}
}

Contributors

Pacific-i64

38 commits

BP

AETHORIA-AI/TR-HASH-MoE-200M-160B-SFT

Model

5

stars

40

commits

3

linked in READMEs

Aug 28, 2026

updated

conversational
custom-code
custom_code
full-parameter-finetuning
gqa
mixture-of-experts
pytorch
safetensors
supervised-finetuning
text-generation
transformers
tr-hash
tr_hash_moe

README

TR-HASH MoE 200M — 160B-source Full SFT (32K release)

Paper

Deterministic multi-hash routing supports long-horizon training in a compact language model

Preprint (DOI) · Model weights · Code

Full-parameter instruction SFT of AETHORIA-AI/TR-HASH-MoE-200M-160B-Refinement on the audited AETHORIA-AI/TR-HASH-MoE-200M-SFT-v2-300K. This release is not LoRA or QLoRA: all 201.2M model parameters were trainable.

The root model.safetensors is the epoch 3 / step 5,982 checkpoint, selected for the strongest full-PIQA result and lowest held-out SFT loss across the three epochs.

Tokenizer: use AETHORIA-AI/TR-HASH-Tokenizer-32K with this 32,000-token model.

Results

EpochStepHeld-out SFT lossSFT pplPIQA accPIQA acc_norm
11,9940.9909432.6967.90%68.93%
23,9880.9639122.6267.85%68.82%
35,9820.9596172.6168.01%69.10%

Released checkpoint benchmark panel

BenchmarkSplitExamplesAccuracyAccuracy (length-normalized)Evaluation backend
PIQAvalidation1,83868.01%69.10%PyTorch FP16, custom Triton
ARC-Easytest2,37657.24%52.86%MLX FP16
ARC-Challengetest1,17227.13%29.01%MLX FP16
ARC Combined (micro)test3,54847.29%44.98%Derived from both ARC test splits
HellaSwagvalidation10,04233.21%38.74%MLX FP16

All benchmark evaluations use zero-shot causal continuation log-likelihood, no chat template and a maximum sequence length of 2,048. Accuracy selects the choice with the highest total continuation log-likelihood; the normalized metric selects by mean continuation log-likelihood per scored token. PIQA was evaluated from the native epoch-3 checkpoint. ARC and HellaSwag were evaluated from an FP16 MLX conversion of the same root F32 SafeTensors weights. ARC Combined is the micro-average over all 3,548 ARC-Easy and ARC-Challenge test examples, not the arithmetic mean of the two percentages. Machine-readable reports are published under reports/sft-v2-300k/.

Compact-model comparison

Combined ARC comparison from 124M to 774M parameters

ModelParametersPIQA accARC-Easy accARC-Challenge accCombined ARC accHellaSwag acc
TR-HASH MoE 200M Full SFT201.2M68.01%57.24%27.13%47.29%33.21%
GPT-2 Large774M53.11%21.76%42.76%
Pythia-410M410M52.02%21.42%41.91%
GPT-2 Medium355M49.16%21.67%40.08%
OPT-350M350M43.98%20.82%36.33%
GPT-2 Small124M62.89%43.81%19.03%≈35.63%28.92%
OPT-125M125M63.00%43.60%19.10%≈35.51%29.20%
Pythia-160M160M62.73%43.52%18.77%35.34%

Combined ARC is weighted by the public test-set sizes (2,376 ARC-Easy and 1,172 ARC-Challenge examples). GPT-2 Small and OPT reference scores come from the AMD-LLM lm-evaluation-harness comparison; their combined values are approximate because the published component scores are rounded. Pythia-160M uses EleutherAI's official zero-shot result. GPT-2 Medium, GPT-2 Large and Pythia-410M were evaluated locally in MLX FP16 with the same causal-choice evaluator as TR-HASH. OPT-350M used the identical prompt and scoring formula in PyTorch MPS FP16 because MLX does not implement the OPT architecture. The 124M–160M references were reported through lm-evaluation-harness, so that part of the comparison is informative rather than a claim of bit-identical evaluation runtimes.

Training recipe

SettingValue
MethodFull-parameter supervised fine-tuning
SourceRefinement step 8,156 (about 162B prior token exposures)
Dataset300,000 train / 3,000 held-out examples
Tokenized corpus202,948,693 train tokens; no truncation
SupervisionFinal assistant turn only; prior assistant turns masked
Epochs3
Context2,048 tokens
TokenizerTR-HASH-Tokenizer-32K; EOS </s> (ID 0)
OptimizerAdamW, betas 0.9 / 0.95, weight decay 0.1
LR2e-5 peak, 3% warmup, continuous cosine decay
PrecisionBF16 training
Root SafeTensors precisionfloat32
KernelsLiger required; custom Triton enabled

Architecture and loading

201.2M parameters, 16 decoder layers, GQA (14 query heads / 2 KV heads), four stored deterministic token-ID-routed experts with top-2 activation, an always-on shared SwiGLU path and tied embeddings. The persisted multi-hash routing tables are part of the checkpoint.

The repository includes an autonomous Transformers adapter. Load it with:

from transformers import AutoModelForCausalLM, AutoTokenizer

repo = "AETHORIA-AI/TR-HASH-MoE-200M-160B-SFT"
tokenizer = AutoTokenizer.from_pretrained(repo, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(repo, trust_remote_code=True)

License

The released model weights and repository code are licensed under CC BY-NC 4.0 as reproduced in LICENSE. Commercial use is not permitted under this license. Source datasets retain their own licenses and terms; see the dataset manifest for the per-source audit.

Citation

@article{Peyriguere_2026,
  title     = {Deterministic multi-hash routing supports long-horizon training in a compact language model},
  author    = {Peyriguere, Boris},
  year      = {2026},
  month     = aug,
  publisher = {Springer Science and Business Media LLC},
  doi       = {10.21203/rs.3.rs-10788774/v1},
  url       = {https://doi.org/10.21203/rs.3.rs-10788774/v1}
}

Contributors

Pacific-i64

38 commits

BP