GitLab

This replaces symbols in tf.keras.mixed_precision.experimental with the corresponding nonexperimental symbols. In some cases, passing a Policy is replaced with passing a policy name for conciseness.

Additionally, for the Shakespeare model, the loss_scale flag is removed, since supporting it with the nonexperimental API is slightly more verbose and it is recommended users use the default loss scale.

PiperOrigin-RevId: 368123944

PiperOrigin-RevId: 361957289

PiperOrigin-RevId: 361181257

PiperOrigin-RevId: 360529962

PiperOrigin-RevId: 360522428

PiperOrigin-RevId: 350170448

PiperOrigin-RevId: 345773699

PiperOrigin-RevId: 345261173

PiperOrigin-RevId: 343410095

PiperOrigin-RevId: 339090333

PiperOrigin-RevId: 338786173

Shift all attribute assignments in functional subclass models below the super().__init__ call, and enable setattr tracking on all functional subclasses.

PiperOrigin-RevId: 338759531

Soon, AutoCastVariable.dtype will refer to the variable dtype, not the compute
dtype. This change stops using AutoCastVariable.dtype.

PiperOrigin-RevId: 337670544

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

PiperOrigin-RevId: 336910013

PiperOrigin-RevId: 334829363

PiperOrigin-RevId: 333878584

Remove return_all_encoder_outputs argument from keras_nlp.BertEncoder as it is unused and we will always return all encoder outputs in the returned dict.

PiperOrigin-RevId: 332802761

PiperOrigin-RevId: 332781597

PiperOrigin-RevId: 332557462

PiperOrigin-RevId: 332506544

PiperOrigin-RevId: 332501400

PiperOrigin-RevId: 332386511

PiperOrigin-RevId: 332122408

PiperOrigin-RevId: 331885219

PiperOrigin-RevId: 331676499

PiperOrigin-RevId: 331634511

PiperOrigin-RevId: 331620370

PiperOrigin-RevId: 331173006

PiperOrigin-RevId: 330754739

PiperOrigin-RevId: 330397061

PiperOrigin-RevId: 329568227

PiperOrigin-RevId: 328872353

PiperOrigin-RevId: 328674302

PiperOrigin-RevId: 328269066

PiperOrigin-RevId: 328269066