This repository was archived by the owner on Mar 31, 2025. It is now read-only.
This repository was archived by the owner on Mar 31, 2025. It is now read-only.
[Jax 0.4.27] AttributeError: arr.device_buffers has been deprecated. #270
Description
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]