SIGN IN SIGN UP

[Pallas:TPU] Add support for int4 to explicit MXU API for TPUv6e and TPUv5*.

PiperOrigin-RevId: 970018499
J
jax authors committed
c0f371a1797f1a1ea7cc1ed2ca0797d856b63e1c
Parent: 2380871