https://github.com/google/jax
Raw File
Tip revision: 3e0610f549a9d3143339a48bdc33e165037e8a85 authored by Yash Katariya on 01 April 2024, 17:53:29 UTC
Expose `fully_replicated_shard` as a method on jax.Array. This is being done to deprecate `.addressable_data` and it's confusing semantics.
Tip revision: 3e0610f
.bazelversion
6.1.2
back to top