jax

9d1c6a04 - [Pallas][TPU kernel interpreter] Make behaviour of `LoggingInfo` TPU/GPU specific.

[Pallas][TPU kernel interpreter] Make behaviour of `LoggingInfo` TPU/GPU specific. Note that currently `LoggingInfo`/`GPULoggingInfo` is not used by the GPU kernel interpreter. PiperOrigin-RevId: 881375009