Team Ai
Modelpublic

AETHORIA-AI/TR-HASH-MoE-200M-130B-Checkpoints

sourceHugging Facecc-by-nc-4.0updated 2mo agoView on Hugging Face
2likes1.2kdownloads
Model Card

TR-HASH MoE 200M — 70B Unique / 130B Replay Checkpoints

Status: training complete. This is the raw checkpoint backup repository, not the finished model release. It exists as a safety net and a record of the training trajectory — token-pack folders include model weights and resumable optimizer/scheduler state.

  • —Checkpoints are uploaded automatically at token-pack boundaries and on clean/interrupted shutdown by `scripts/sync_checkpoints_to_hf.py`.
  • —Folder names follow {tag}_{step} (token_pack_NNN_STEP, final_STEP, interrupted_STEP).
  • —The PIQA sweep below is exploratory, not a full validated evaluation suite. Do not treat an individual checkpoint as a finished model release.
  • —The architecture config is tracked at `model_config.yaml` in this repo (not embedded in checkpoint.pt).

Exploratory zero-shot checks (not a full evaluation)

Informal checks across the final token-pack checkpoints, via `scripts/convert_to_mlx.py` + `scripts/eval_mlx_zero_shot.py`, zero-shot causal-log-likelihood scoring, no chat template. Every row uses the same tokenizer, MLX FP16 inference path, PIQA validation split, and 1,838 examples:

CheckpointTokens trainedLearning ratePIQA accPIQA acc_norm
token_pack_032_132239104.00B3.00e-40.67360.6763
token_pack_033_136371107.25B2.91e-40.67740.6746
token_pack_034_140504110.50B2.62e-40.67680.6703
`token_pack_035_144636`113.75B2.19e-40.67410.6801
token_pack_036_148769117.00B1.67e-40.67680.6774
token_pack_037_152901120.25B1.14e-40.66760.6697
token_pack_038_157034123.50B7.01e-50.65780.6638
token_pack_039_161166126.75B4.04e-50.65670.6529
token_pack_040_165298 / final130.00B3.00e-50.65450.6561

token_pack_035_144636 has the highest acc_norm in this sweep. The trajectory is not monotonic, and PIQA alone must not be used as a complete model-quality or checkpoint-selection criterion.

The finished base model is published separately as AETHORIA-AI/TR-HASH-MoE-200M-130B.

See Complexity Framework for the training code.