Slice BERT output with a TF Op layer instead of a Lambda layer.
Slice BERT output with a TF Op layer instead of a Lambda layer.

Lambda layers are fundamentally non-portable since they serialize
Python bytecode.

PiperOrigin-RevId: 336910013

keras-team/keras-nlp,keras-team/keras-nlp