penzai.models# Submodules penzai.models.simple_mlp A simple multi-layer perceptron. penzai.models.transformer Basic transformer backbone with variants.