v0.1.4

What's Changed

• Extract lambda to a named function to ensure cache hits. by @copybara-service[bot] in #127
• Always write array metadata if self._array_metadata_store is not None. by @copybara-service[bot] in #125
• Add "Shared Pathways Service" for Pathways-on-Cloud by @copybara-service[bot] in #128
• Exposes pathwaysutils.profiling._start_trace_from_profile_request as pathwaysutils.experimental.profiling.start_trace. by @copybara-service[bot] in #141
• Patch internal JAX profiler functions (enabling jax.profiler.trace) and add a test for jax.profiler.trace. by @copybara-service[bot] in #140
• Expose _split_by_mesh_axis directly in pw_jax. by @copybara-service[bot] in #147
• Use Pathways _transfer_to_sharding for resharding in experimental reshard API. by @copybara-service[bot] in #148
• Allow specifying a custom proxy job name in _ISCPathways by @copybara-service[bot] in #157

v0.1.3

What's Changed

• Update the github action for PyPI by @lukebaumann in #105
• Expose is_pathways_backend_used in pathwaysutils. 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 and CloudPathwaysArrayHandler

Full Changelog: v0.1.2...v0.1.3

v0.1.2

What's Changed

• Updates to Pathways orbax handler. In #81 #89 #90
• Improvements to pathwaysutils/elastic. In #72 #84 #85 #86 #99 #100 #102
• Handle JAX API interface compatibility. In #88 #93 #95
• Adding jax version to the github actions matrix by @lukebaumann in #94
• Added LRU cache and tests to pathwaysutils. by @copybara-service[bot] in #83
• Moving initialization logic into its own module by @copybara-service[bot] in #96

Full Changelog: v0.1.1...v0.1.2

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 from import 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.