jax

ccb5f92f - [Mosaic GPU][NFC] Add null checks before nullable pointers are dereferenced.

[Mosaic GPU][NFC] Add null checks before nullable pointers are dereferenced. `collective_params` and `clique_requests` are expected to be non-null when `ModuleUsesCollectiveMetadata` returns `true`. Otherwise, these can be null. PiperOrigin-RevId: 860152056