Update pathways.experimental.reshard so that PyTrees with arrays that have different device sets can be resharded.#113

copybara-service[bot] merged 1 commit intomainAI-Hypercomputer/pathways-utils:mainfrom

test_811039183AI-Hypercomputer/pathways-utils:test_811039183Copy head branch name to clipboard