Checkpoints

Contents

Checkpoints#

Kaggle Hub#

Gemma models are available on Kaggle Hub for various formats:

To manually download the model:

  • Select one of the Flax model variations

  • Click the “Download” button to download the model archive

  • Extract the model files from the archive

The archive contains both the model weights and the tokenizer, like:

2b/              # Directory containing model weights
tokenizer.model  # Tokenizer

To programmatically download the model:

import kagglehub

kagglehub.login()

weights_dir = kagglehub.model_download(f'google/gemma-3/flax/gemma3-4b-it')

v4#

NOTE: Gemma 4 Flax checkpoints will be updated on Kaggle when possible.

v3#

Size

Variant

Quantized

Name

1B

Pre-trained

google/gemma-3/flax/gemma3-1b

: : (PT) : : :

4B

Pre-trained

google/gemma-3/flax/gemma3-4b

: : (PT) : : :

12B

Pre-trained

google/gemma-3/flax/gemma3-12b

: : (PT) : : :

27B

Pre-trained

google/gemma-3/flax/gemma3-27b

: : (PT) : : :

1B

Instruction

google/gemma-3/flax/gemma3-1b-it

: : Tuned (IT) : : :

4B

Instruction

google/gemma-3/flax/gemma3-4b-it

: : Tuned (IT) : : :

12B

Instruction

google/gemma-3/flax/gemma3-12b-it

: : Tuned (IT) : : :

27B

Instruction

google/gemma-3/flax/gemma3-27b-it

: : Tuned (IT) : : :

1B

Instruction

Y

google/gemma-3/flax/gemma3-1b-it-int4

: : Tuned (IT) : : :

4B

Instruction

Y

google/gemma-3/flax/gemma3-4b-it-int4

: : Tuned (IT) : : :

12B

Instruction

Y

google/gemma-3/flax/gemma3-12b-it-int4

: : Tuned (IT) : : :

27B

Instruction

Y

google/gemma-3/flax/gemma3-27b-it-int4

: : Tuned (IT) : : :

v2#

Size

Variant

Name

2.6B

Pre-trained (PT)

google/gemma-2/flax/gemma2-2b

9B

Pre-trained (PT)

google/gemma-2/flax/gemma2-9b

27B

Pre-trained (PT)

google/gemma-2/flax/gemma2-27b

2.6B

Instruction Tuned (IT)

google/gemma-2/flax/gemma2-2b-it

9B

Instruction Tuned (IT)

google/gemma-2/flax/gemma2-9b-it

27B

Instruction Tuned (IT)

google/gemma-2/flax/gemma2-27b-it