G0-nano-base

A 62M-parameter GPT trained completely from scratch on a single 8GB-RAM device (an NVIDIA Jetson) — no cloud cluster, no multi-GPU node. This is the base (pretrained, non-instruct) checkpoint: it completes text, it does not follow instructions. See G0-nano-instruct for the chat-tuned version.

The point of this model isn't to compete with billion-parameter LLMs — a 62M model is physically capped at roughly 15.5MB of factual knowledge (~2 bits/parameter, Allen-Zhu & Li, ICLR'25), regardless of training method. The point is that it was trained end-to-end, pretraining included, within an 8GB memory budget.

Architecture

Llama-style decoder-only transformer:

Parameters 62.1M (embeddings shared with LM head)
Layers 12
Hidden size 640
Attention Grouped-Query Attention, 10 query heads / 2 KV heads, head_dim 64
Position encoding RoPE (θ=10000)
Feed-forward SwiGLU, hidden 1728
Normalization RMSNorm
Context length 1024 tokens
Vocabulary 16,384 (SentencePiece, trained from scratch)
Precision fp32

Trained on ~1.5B tokens of English web/book text (FineWeb-Edu, BookCorpus, OpenWebText, PG-19, WikiHow).

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer

tok = AutoTokenizer.from_pretrained("AZERDSQ/G0-nano-base", trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained("AZERDSQ/G0-nano-base", trust_remote_code=True)

inputs = tok("The city of Paris is", return_tensors="pt")
out = model.generate(**inputs, max_new_tokens=50, do_sample=True, top_k=50, temperature=0.8)
print(tok.decode(out[0]))

trust_remote_code=True is required — this is a custom architecture (GQA + RoPE + SwiGLU), not one of the built-in transformers model types.

Limitations

  • 62M parameters caps factual knowledge hard — expect confident, fluent, frequently wrong completions on anything knowledge-dense.
  • 1024-token context.
  • English only.
  • The custom tokenizer/model code does not handle padded batched inference — single-sequence generation only.
  • Base model: raw next-token prediction, no chat formatting. It will continue text, not answer questions.

License

Apache 2.0. Weights only — this release does not include training code or data pipelines.

Downloads last month
26
Safetensors
Model size
62.1M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Paper for AZERDSQ/G0-nano-base