jax

3c2d0db3 - [pallas:sc] Removed support for `pl.get_global`

[pallas:sc] Removed support for `pl.get_global` It is no longer used, since we can use `pl.core_map` with `scratch_shapes=` to allocate in `VMEM_SHARED` instead. PiperOrigin-RevId: 871777871