gemma-4-31B | NVIDIA NeMo AutoModel

Gemma-4-31B has a checked-in NeMo AutoModel recipe for text generation. The Hugging Face configuration declares the Gemma4ForConditionalGeneration architecture.

Fine-Tune Gemma-4-31B

Follow the installation instructions, then review the checked-in recipe and its declared topology below before choosing a local or cluster launcher.

Use the Slurm launcher guide for the multi-node run.

Choose a Workflow

Goal	Start Here
Run the primary recipe for this model	Use the recipe configuration.
Try another checked-in recipe for this model	Use the alternate recipe.

Configuration

Setting	Configuration
Hardware	-
Strategy	FSDP2; tp_size=1, cp_size=1
Nodes	4
Features	Checkpointing (enabled=true), Activation checkpointing (activation_checkpointing=true)
Advanced	torch.bfloat16

More Recipes for This Model

gemma4_31b_base_coderforge_cp8_64k_1e5_800steps.yaml

Model Reference

Model Architecture

Property	Value
Task	Text generation
Architecture	`Gemma4ForConditionalGeneration`

Available Models

google/gemma-4-31B

LLM fine-tuning guide