jax-cfd icon indicating copy to clipboard operation
jax-cfd copied to clipboard

Published 20 hours ago • verified publisher icon google

Exception has occurred: KeyError 'experimental_xmap_spmd_lowering'

Open MCSchulz-git opened this issue 1 year ago • 0 comments

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

MCSchulz-git avatar May 12 '23 09:05 MCSchulz-git