add set_nan_for_oob_id flag to the Embedding layer to map embedding for out-of-boundary ids to nan to help debug. It is set to False by default.
add set_nan_for_oob_id flag to the Embedding layer to map embedding for out-of-boundary ids to nan to help debug. It is set to False by default.

PiperOrigin-RevId: 455274859

google/praxis,google/praxis