Exception has occurred: KeyError 'experimental_xmap_spmd_lowering'
Open
- Dominant language
- Jupyter Notebook
- Stars
- 964
- Forks
- 142
- Avg merge
- 3h 9m
- Merged PRs (30d)
- 1
Description
When running layers_util_test.py with jax[CPU] version later than 0.4.1, it throws a key error 'experimental_xmap_spmd_lowering'.
Contributor guide
Assessment
This issue has not been assessed yet.