weight-sync(jax): fail-loud runtime-derived tile-representability guard - #666

weight-sync(jax): fail-loud runtime-derived tile-representability guard#666

lokic233 wants to merge 1 commit into

google:maingoogle/tpu-sync:mainfrom

lokic233:pr-a-tile-representability-guardlokic233/tpu-sync:pr-a-tile-representability-guardCopy head branch name to clipboard