skt/A.X-K1
✓ Same architecture, smaller
Every module class, layer kind and checked property of the original is in the fixture.
Model tester (advisory)
The model's own small test config in its ModelTester (29 literal values, 13 set by the model's own tester, the rest generic base-tester defaults). Testers are usually representative but not always, so these notes are for review, not rules.
The fixture covers every per-layer kind the tester sets, and shrinks every size it does.
Layer stacks
One square per layer, colored by kind: the layer's module classes and type attributes. Hover a square for its index and kind.
model.layers
| Kind | Original | Fixture | |
|---|---|---|---|
| no Experts | 1 | 1 | ✓ kept |
| Experts, MoE, TopkRouter | 60 | 1 | ✓ kept |
Where the parameters are
Share of each model's parameters per component. Vocabulary and patch sizes are kept, so embeddings usually take a larger share of the fixture.
| Component | Original | Share | Fixture | Share |
|---|---|---|---|---|
| model.layers | 516.6B | 99.5% | 76.4K | 0.4% |
| model.embed_tokens | 1.2B | 0.2% | 10.5M | 49.8% |
| lm_head | 1.2B | 0.2% | 10.5M | 49.8% |
| model.norm | 7.2K | 0.0% | 64 | 0.0% |
Preserved properties
| Config | Property | Original | Fixture | |
|---|---|---|---|---|
| model | attention style | MHA (64 heads, 64 KV) | MHA (4 heads, 4 KV) | ✓ kept |
| model | MoE routing | top-8 of 192 experts | top-2 of 4 experts | ✓ kept |
| model | RoPE type | yarn | yarn | ✓ kept |
| model | model_type | axk1 | axk1 | ✓ kept |
| model | vocab_size | 163840 | 163840 | ✓ kept |
| model | tie_word_embeddings | False | False | ✓ kept |
| model | hidden_act | silu | silu | ✓ kept |
| model | max_position_embeddings | 131072 | 131072 | ✓ kept |
Module classes
Instances of each class in the built model (prefix AXK1 omitted).
| Class | Original | Fixture |
|---|---|---|
| Linear | 489 | 17 |
| RMSNorm | 305 | 10 |
| SiLUActivation | 121 | 3 |
| Attention | 61 | 2 |
| DecoderLayer | 61 | 2 |
| MLP | 61 | 2 |
| Experts | 60 | 1 |
| MoE | 60 | 1 |
| TopkRouter | 60 | 1 |
| ForCausalLM | 1 | 1 |
| Model | 1 | 1 |
| RotaryEmbedding | 1 | 1 |
| Embedding | 1 | 1 |
| ModuleList | 1 | 1 |
Config changes
17 fields changed, 34 unchanged.
| Field | Original | Fixture |
|---|---|---|
| head_dim | 64 | 8 |
| hidden_size | 7168 | 64 |
| intermediate_size | 18432 | 128 |
| kv_lora_rank | 512 | 16 |
| mlp_layer_types | 61 entries: 60× sparse, 1× dense | ["dense", "sparse"] |
| moe_intermediate_size | 2048 | 32 |
| n_group | 8 | 1 |
| n_routed_experts | 192 | 4 |
| num_attention_heads | 64 | 4 |
| num_experts_per_tok | 8 | 2 |
| num_hidden_layers | 61 | 2 |
| num_key_value_heads | 64 | 4 |
| q_lora_rank | 1536 | 16 |
| qk_nope_head_dim | 128 | 16 |
| qk_rope_head_dim | 64 | 8 |
| topk_group | 4 | 1 |
| v_head_dim | 128 | 16 |