版本发布 8
## What's Changed * Refactor elasticity retry logic into a reusable private method. by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/177 * Add a name parameter to the watchdog context manager. by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/199 * Use abstract types for return type hints in pathwaysutils. by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/198 * Update environment variables for JAX backend by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/203 * Enable Pathways profiling with jax.profiler.ProfileOptions. by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/201 * Refactor elastic retry decorators into a single elastic_retry method. by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/202 * Fix JaxRuntimeError during profiler stop_trace with profile options by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/205 * Add CLI mode to Shared Pathways Service by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/200 **Full Changelog**: https://github.com/AI-Hypercomputer/pathways-utils/compare/v0.1.6...v0.1.7
## What's Changed * Extract lambda to a named function to ensure cache hits. by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/127 * Always write array metadata if self._array_metadata_store is not None. by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/125 * Add "Shared Pathways Service" for Pathways-on-Cloud by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/128 * Exposes `pathwaysutils.profiling._start_trace_from_profile_request` as `pathwaysutils.experimental.profiling.start_trace`. by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/141 * Patch internal JAX profiler functions (enabling `jax.profiler.trace`) and add a test for `jax.profiler.trace`. by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/140 * Expose `_split_by_mesh_axis` directly in `pw_jax`. by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/147 * Use Pathways `_transfer_to_sharding` for resharding in experimental reshard API. by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/148 * Allow specifying a custom proxy job name in _ISCPathways by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/157
## What's Changed * Updates to Pathways orbax handler. In https://github.com/AI-Hypercomputer/pathways-utils/pull/81 https://github.com/AI-Hypercomputer/pathways-utils/pull/89 https://github.com/AI-Hypercomputer/pathways-utils/pull/90 * Improvements to pathwaysutils/elastic by @copybara-service[bot] inhttps://github.com/AI-Hypercomputer/pathways-utils/pull/72 https://github.com/AI-Hypercomputer/pathways-utils/pull/84 https://github.com/AI-Hypercomputer/pathways-utils/pull/85 https://github.com/AI-Hypercomputer/pathways-utils/pull/86 https://github.com/AI-Hypercomputer/pathways-utils/pull/99 https://github.com/AI-Hypercomputer/pathways-utils/pull/100 https://github.com/AI-Hypercomputer/pathways-utils/pull/102 * Handle JAX API interface compatibility by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/88 https://github.com/AI-Hypercomputer/pathways-utils/pull/93 https://github.com/AI-Hypercomputer/pathways-utils/pull/95 * Adding jax version to the github actions matrix by @lukebaumann in https://github.com/AI-Hypercomputer/pathways-utils/pull/94 * Added LRU cache and tests to pathwaysutils. by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/83 * Moving initialization logic into its own module by @copybara-service[bot] in https://github.com/AI-Hypercomputer/pathways-utils/pull/96 **Full Changelog**: https://github.com/AI-Hypercomputer/pathways-utils/compare/v0.1.1...v0.1.2
## 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**: https://github.com/AI-Hypercomputer/pathways-utils/compare/v0.1.0...v0.1.1
## What's Changed * Introduce `pathwaysutils.initialize()` to remove relying on side-effects from `import pathwaysutils`. by @copybara-service in https://github.com/AI-Hypercomputer/pathways-utils/pull/47 * Adding debugging utilities by @copybara-service in https://github.com/AI-Hypercomputer/pathways-utils/pull/57 * Adding an elastic manager and reshard modules by @copybara-service in https://github.com/AI-Hypercomputer/pathways-utils/pull/58 * Update README for Colocated Python Sidecar. by @copybara-service in https://github.com/AI-Hypercomputer/pathways-utils/pull/60 **Full Changelog**: https://github.com/AI-Hypercomputer/pathways-utils/compare/v0.0.8...v0.1.0
* Disabled JAX's compilation cache * Updated Orbax handler to use bulk APIs * Updates to support JAX 0.5.0 **Full Changelog**: https://github.com/AI-Hypercomputer/pathways-utils/compare/v0.0.7...v0.0.8
* Decreased logging severity for most logs * Persistence enabled * General argument type fixes
# Changes * Decreased logging severity for most logs * Persistence enabled * General argument type fixes