jax

f311e81c - Added `is_signed` to `mgpu.FragmentedArray`

Added `is_signed` to `mgpu.FragmentedArray` The registers within a fragmented array always use signless types, and instead the signedness is tracked on the fragmented arrays itself (i.e. in Python). PiperOrigin-RevId: 677776009