penzai.example_models# Submodules penzai.example_models.gemma Modular implementation of the Gemma model architecture. penzai.example_models.simple_mlp A simple multi-layer perceptron.