jax

46d427b6 - [Mosaic GPU][NFC] Restrict valid `packing` values.

[Mosaic GPU][NFC] Restrict valid `packing` values. TMEM references are either unpacked (packing = 1) or fully packed (packing = 32 // element_type_bit_width). PiperOrigin-RevId: 827462237