jax

0ec03dbd - [XLA:Python] Fix __cuda_array_interface__.

[XLA:Python] Fix __cuda_array_interface__. Adds a test for __cuda_array_interface__ that does not depend on cupy. Fixes https://github.com/google/jax/issues/16440 PiperOrigin-RevId: 541965361