What's Changed
- Update the github action for PyPI by @lukebaumann in #105
- Expose
is_pathways_backend_used
inpathwaysutils
. by @copybara-service[bot] in #107 - Treat additional error types as potential slice down issues. by @copybara-service[bot] in #109
- Adding split_by_mesh_axis to experimental for use by a new experimental reshard. by @copybara-service[bot] in #112
- Update pathways.experimental.reshard so that PyTrees with arrays that have different device sets can be resharded. by @copybara-service[bot] in #113
- Update GitHub action unittest matrix by @lukebaumann in #111
- Handle new style PRNG keys in
reshard
andCloudPathwaysArrayHandler
Full Changelog: v0.1.2...v0.1.3