acme icon indicating copy to clipboard operation
acme copied to clipboard

Published 20 hours ago • verified publisher icon google-deepmind

Support JAX Array

Open cemlyn007 opened this issue 2 years ago • 0 comments
trafficstars

Happy New Year guys!

Are you guys planning on supporting JAX Array and removing all uses of DeviceArray, ShardedDeviceArray, and GlobalDeviceArray types in JAX?

cemlyn007 avatar Jan 11 '23 11:01 cemlyn007