You can deploy this model and start earning money today!

deqing/
convergent-mamba2-300M-muon-original

Name: convergent-mamba2-300M-muon-original
Availability: InStock

HuggingFaceFW/fineweb-edu

mit

endpoints_compatible

Views

No views yet

convergent-mamba2-300M-muon-original

A 300M-parameter language model trained from scratch on FineWeb-Edu 10BT (~9.4B tokens, 1 epoch) as part of the Convergent Evolution project, which investigates how Fourier features emerge in LLM number embeddings.

Model details


Architecture	Mamba-2 SSM (state-space model)
Parameters	~300M
Optimizer	Muon (for 2D weights) + AdamW (for embeddings/bias/norm)
Data perturbation	standard (unperturbed) text
Training data	FineWeb-Edu sample-10BT (~9.4B tokens)
Context length	1024
Tokenizer	Llama 3 (128K vocab)
Batch size	512 sequences

Training dynamics

Intermediate checkpoints are saved as branches: tokens-200M, tokens-400M, ..., tokens-9.6B.

python

from transformers import AutoModelForCausalLM

# Load final checkpoint
model = AutoModelForCausalLM.from_pretrained("deqing/convergent-mamba2-300M-muon-original")

# Load intermediate checkpoint (e.g., at 1B tokens)
model = AutoModelForCausalLM.from_pretrained("deqing/convergent-mamba2-300M-muon-original", revision="tokens-1B")

Citation

Paper forthcoming.