deepseek-v4-mini-3B-init

DeepSeek-V4 Mini (3B) — randomly-initialized architecture replica

Faithful small-scale (~3.2B total / ~1.10B activated per token) replica of the DeepSeek-V4 architecture, sized to be trainable on rented GPUs and to map cleanly onto the full-scale V4-Flash dimensions for weight slicing.

This is a randomly-initialized scaffold — generates noise. Its purpose:

• reference architecture for ablation / hyperparameter-search experiments
• target for weight transfer / slicing from real V4-Pro / V4-Flash

Architecture summary

Quick start

from huggingface_hub import login, snapshot_download
login()  # private repo

local = snapshot_download(repo_id="kshitijthakkar/deepseek-v4-mini-3B-init")

import sys, os
sys.path.insert(0, os.path.join(local, "code"))
import deepseek_v4  # registers DeepseekV4{Config,ForCausalLM} with HF auto classes

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer

tok = AutoTokenizer.from_pretrained(local)
model = AutoModelForCausalLM.from_pretrained(local, torch_dtype=torch.bfloat16)
model.eval()

ids = tok.apply_chat_template(
    [{"role": "user", "content": "Hello"}],
    return_tensors="pt", add_generation_prompt=True, return_dict=True,
)
with torch.no_grad():
    out = model(input_ids=ids["input_ids"])
print(out.logits.shape)

Components implemented

mHC (Sinkhorn-Knopp) · CSA + Lightning Indexer · HCA · pure sliding-window · Shared-KV MQA + grouped output projection (per-group wo_a) · partial RoPE + output -i rotation · attention sink · DeepseekMoE with sqrt(softplus) routing · hash-routed early layers · clamped SwiGLU · MTP head · YaRN.

Every component is bit-equivalent in math to the official inference/model.py

• kernel.py:hc_split_sinkhorn (FP4/FP8 quantization and Hadamard rotation are skipped — those are inference optimizations, not architecture).

Citation

@misc{deepseek_v4_2026,
  author = {DeepSeek-AI},
  title  = {DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligence},
  year   = {2026},
  url    = {https://huggingface.co/deepseek-ai/DeepSeek-V4-Flash}
}