Downloads · 30 days
0
anlp-a2-part1-moe_shared
neemon/anlp-a2-part1-moe_shared
anlp-a2-part1-moe_shared is a translation model from neemon. Use it when you need text moved from one language to another. It is set up for pytorch. The card lists the license as mit.
Decoder-only transformer with 1 shared expert always active plus 3 routed experts, top-1 routing.
Access
Public
Updated Oct 1, 2026
Repo size
499 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt166 MB · 98%
At a glance
- Task
- Translation
- Library
- pytorch
- License
- mit
- Access
- Public
- Created
- Sep 14, 2026
- Updated
- Oct 1, 2026
- SHA
- 04361443
Try a prompt
Model card
From the Hugging Face model README
ANLP Assignment 2 (part1): moe_shared
Decoder-only transformer with 1 shared expert always active plus 3 routed experts, top-1 routing.
Trained from scratch for IIIT-H Advanced NLP, Monsoon 2026. Architecture, training code and evaluation live in the assignment repository. This repository holds the checkpoint and the tokenizer it was trained with, which every model in the assignment shares.
Architecture
| Setting | Value |
|---|---|
d_model | 512 |
n_layers | 8 |
n_heads | 8 |
n_kv_heads | 8 |
n_ctx | 256 |
vocab_size | 32000 |
ffn | moe_shared |
d_ff | 512 |
n_routed_experts | 3 |
n_shared_experts | 1 |
top_k | 1 |
norm | rmsnorm |
Parameters
| Count | Value |
|---|---|
| Total | 41,603,584 |
| Active per token | 33,198,592 |
| Feed-forward (total / active) | 16,809,984 / 8,404,992 |
Test set results
| Metric | Value |
|---|---|
| Perplexity (both directions) | 9.58 |
| BLEU vi->en | 34.42 |
| BLEU ja->en | 24.81 |
| BLEU mean | 29.61 |
Training
- Scored target tokens: 33,393,292
- Best validation loss: 2.0135
- Best validation perplexity: 7.49
Loading
import torch
payload = torch.load("model.pt", map_location="cpu", weights_only=False)
state_dict, config = payload["model"], payload["config"]
from tokenizers import Tokenizer
tokenizer = Tokenizer.from_file("tokenizer.json")
- Task
- Translation
- Library
- pytorch
- License
- mit
- Languages
- en, vi, ja
- Created
- Sep 14, 2026
- Updated
- Oct 1, 2026