google-deepmind / google-deepmind/acme
Support JAX Array
Open
- Dominant language
- Python
- Stars
- 4.1k
- Forks
- 553
- PR merge metrics
- No merged PRs in 30d
Description
Happy New Year guys!
Are you guys planning on supporting JAX Array and removing all uses of `DeviceArray`, `ShardedDeviceArray`, and `GlobalDeviceArray` types in JAX?
Contributor guide
Assessment
This issue has not been assessed yet.