JetMoE 8B β€” RL-MPQ Balanced

Standalone RL-MPQ (Reinforcement Learning Mixed-Precision Quantization) checkpoint for the Balanced scenario β€” a quantized variant of jetmoe/jetmoe-8b.

Field Value
Base model jetmoe/jetmoe-8b
Scenario Balanced
Avg bits / weight 4.5
Compression vs FP16 3.5556Γ—
WikiText-2 PPL 4.6648
Layers 24
Bit distribution {'4': 21, '8': 3}
Format Fake-quant FP16 + rlmpq_policy.json

Collection: RL-MPQ β€” JetMoE 8B β€” all five scenarios for JetMoE 8B.

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer

repo = "AvoCahDoe/jetmoe-8b-rlmpq-balanced"

model = AutoModelForCausalLM.from_pretrained(repo, torch_dtype="float16")
tokenizer = AutoTokenizer.from_pretrained(repo)

Other JetMoE 8B scenarios

Scenario Avg bits Compression WikiText-2 PPL
Aggressive 3.5833 4.4651x 4.6648
Conservative 5.3333 3.0x 4.6648
Extreme Survival 2.625 6.0952x 4.6648
High Fidelity 6.8333 2.3415x 4.6648

Grouped archive (all scenarios in one repo): AvoCahDoe/jetmoe-8b-rlmpq

Method

  1. Phase 3 β€” PPO agent assigns per-layer bit widths under the Balanced reward target.
  2. Phase 4 β€” Policy replayed on real weights; WikiText-2 perplexity validates quality.
  3. Export β€” Fake-quantized FP16 weights compatible with Hugging Face Transformers.

Files

File Description
config.json Llama architecture + RL-MPQ metadata
model.safetensors Fake-quantized weights
rlmpq_policy.json Per-layer bit-width policy
rlmpq_metrics.json Validation & PPL summary

Citation

@misc{rlmpq_jetmoe_8b_balanced_2026,
  title  = {RL-MPQ Balanced: JetMoE 8B Mixed-Precision Quantization},
  author = {AvoCahDoe},
  year   = {2026},
  url    = {https://huggingface.co/AvoCahDoe/jetmoe-8b-rlmpq-balanced}
}
Downloads last month
43
Safetensors
Model size
9B params
Tensor type
BF16
Β·
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Model tree for AvoCahDoe/jetmoe-8b-rlmpq-balanced

Base model

jetmoe/jetmoe-8b
Finetuned
(7)
this model

Dataset used to train AvoCahDoe/jetmoe-8b-rlmpq-balanced

Collection including AvoCahDoe/jetmoe-8b-rlmpq-balanced