[Jax 0.4.27] AttributeError: arr.device_buffers has been deprecated.

[GaetanLepage]

Since jax 0.4.27, several tests fail with:

    @property
    def device_buffers(self):
>     raise AttributeError(
        "arr.device_buffers has been deprecated. Use [x.data for x in arr.addressable_shards]")
E     AttributeError: arr.device_buffers has been deprecated. Use [x.data for x in arr.addressable_shards]