Reduce Trax layers error messages for less noise and almost the same signal.
Reduce Trax layers error messages for less noise and almost the same signal.

PiperOrigin-RevId: 317047102

google/trax,google/trax