ApX logoApX logo

Yi-9B

Parameters

9B

Context Length

4K

Modality

Text

Architecture

Dense

KV Cache Rate

88 KB/tok

License

Apache 2.0

Release Date

6 Mar 2024

Knowledge Cutoff

Jun 2023

API Pricing (per 1M)

Self-hosted only

Hardware Sizing

Precision

Context

Workload Memory: 7.03 GB

INT4 · 4K context

Base Weights: 4.19 GB

KV Cache: 0.34 GB

Runtime & Graphs: 2.50 GB

Minimum Required Hardware

1x NVIDIA L4 24GB

7.0 / 21.6 GB usable

1x RTX 3060 12GB

7.0 / 10.8 GB usable

Apple M4 24GB

7.0 / 18.0 GB usable

Determine requirements for your specific workload:

Open in VRAM Calculator

Architecture Diagram

Input TokensToken EmbeddingPosition: AbsoluteHidden: 4.1k · Context: 4K · Vocab: 64kx 44 layersRMSNormPre-AttentionMulti-Head Attention32Q / 4KV headsHead dim: 128+RMSNormPre-FFNFeed-Forward NetworkSwiGLUIntermediate: 11k+Final RMSNormOutput Logits

Evaluation Benchmarks

No evaluation benchmarks for Yi-9B available.

Rankings

Overall Rank

-

Coding Rank

-

About Yi-9B

Yi-9B is an efficient dense transformer by 01.AI engineered for technical reasoning, mathematics, and bilingual coding on consumer hardware. Pretrained on 3.9T tokens, it provides a strong balance of analytical depth and local throughput.

Technical Specifications

Attention

Attention Structure

Multi-Head Attention

Attention Heads

32

Key-Value Heads

4

Attention Head Dimension

128

Position Embedding

Absolute Position Embedding

RoPE Theta

10,000

Normalization

RMS Normalization

Activation Function

SwigLU

Dimensions

Hidden Dimension Size

4,096

Number of Layers

44

FFN Intermediate Size (Dense)

11,008

Tokenizer

Vocabulary Size

64,000

About Yi

Yi series models are large language models trained from scratch by 01.AI. Bilingual (English/Chinese), featuring strong performance in language understanding, reasoning, and code generation.


Other Yi Models