AttributeError: module 'jax.interpreters.xla' has no attribute 'DeviceArray'
Open
- Dominant language
- Jupyter Notebook
- Stars
- 10.9k
- Forks
- 1.4k
- PR merge metrics
- No merged PRs in 30d
Description
Hi.
The dependencies should be changed (particularly the `jax` version shouldn't be the latest one `0.4.13`).
Related [issue ](https://github.com/deepmind/optax/issues/533) in another repo.
Tested in [colab](https://colab.research.google.com/github/google/dopamine/blob/master/dopamine/colab/jax_agent_visualizer.ipynb)
Contributor guide
Assessment
This issue has not been assessed yet.