Releases: AI-Hypercomputer/pathways-utils
Releases · AI-Hypercomputer/pathways-utils
v0.1.1
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
v0.1.0
What's Changed
- Introduce
pathwaysutils.initialize()
to remove relying on side-effects fromimport pathwaysutils
. by @copybara-service in #47 - Adding debugging utilities by @copybara-service in #57
- Adding an elastic manager and reshard modules by @copybara-service in #58
- Update README for Colocated Python Sidecar. by @copybara-service in #60
Full Changelog: v0.0.8...v0.1.0
v0.0.8
- Disabled JAX's compilation cache
- Updated Orbax handler to use bulk APIs
- Updates to support JAX 0.5.0
Full Changelog: v0.0.7...v0.0.8
v0.0.7
- Decreased logging severity for most logs
- Persistence enabled
- General argument type fixes
v0.0.6
Changes
- Decreased logging severity for most logs
- Persistence enabled
- General argument type fixes
Initial Public Release
Initial public release of the Pathways-on-Cloud utilities package.