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