MCSchulz-git

Results 1 issues of MCSchulz-git

Exception has occurred: KeyError 'experimental_xmap_spmd_lowering'

When running layers_util_test.py with jax[CPU] version later than 0.4.1, it throws a key error 'experimental_xmap_spmd_lowering'.