def __call__(self, value: NamedArray, **_unused_side_inputs) -> NamedArray:
    """Adds a learned bias to the value."""
    # Elementwise functions broadcast automatically
    return value + self.bias.value
Adds a learned bias to the value.
penzai/nn/linear_and_affine.py