jax

113cb4d2 - [Pallas/Mosaic GPU] Add `__eq__` and `__hash__` implementations for `AbstractRefUnion`.

[Pallas/Mosaic GPU] Add `__eq__` and `__hash__` implementations for `AbstractRefUnion`. Also add a test that would exercise `__eq__` and `__hash__` under lowering with caching; for now, it doesn't exercise these paths. PiperOrigin-RevId: 934456234