Replace DeviceArray
with Array
in the docs
#373
Labels
documentation
Improvements or additions to documentation
DeviceArray
with Array
in the docs
#373
JAX has introduced
jax.Array
, which "subsumesDeviceArray
,ShardedDeviceArray
, andGlobalDeviceArray
" (see https://jax.readthedocs.io/en/latest/jax_array_migration.html). So, if we want to talk about the JAX array type, it should beArray
, notDeviceArray
as we were previously doing in many places in the docs.The text was updated successfully, but these errors were encountered: