What's Changed
- Port the
collect_profile
script from JAX to PathwaysUtils - Remove support for legacy initialize
- Add collect_profile as a script of pathwaysutils
- Make CloudPathwaysArrayHandler compatible with async directory creation feature in orbax
Full Changelog: v0.1.0...v0.1.1