jax

ebe34614 - [Mosaic] Set TensorCore as the default kernel core

[Mosaic] Set TensorCore as the default kernel core Instead of setting the default value in all the callers of `GetCoreTypeOfParentFunc`, do so inside the latter. PiperOrigin-RevId: 774897244