gemma

The Gemma architecture transformer variant.

See the Gemma technical report at https://storage.googleapis.com/deepmind-media/gemma/gemma-report.pdf and the accompanying reference implementation at google-deepmind/gemma.

Functions

gemma_from_pretrained_checkpoint(ckpt_params)

Builds a Gemma model from a pretrained checkpoint.