ai-hypercomputer/pathways-utils issues and pull requests

#69 - Updated some docstrings in `pathwaysutils/init.py`, removed a warning, updated tests, and changed warnings to debug logs.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#68 - Use the info logging level within the elastic manager to reduce log spam.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#67 - Update Sidecar README.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#66 - Update README.md

Pull Request - State: closed - Opened by RoshaniN 4 months ago

#65 - Update README.md

Pull Request - State: open - Opened by ultrons 4 months ago

#64 - Remove support for legacy initialize

Pull Request - State: open - Opened by copybara-service[bot] 4 months ago - 1 comment

#63 - Create Dockerfile for building the custom Colocated Python server image.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#62 - Updating version

Pull Request - State: closed - Opened by lukebaumann 4 months ago

#61 - Small update to the colocated python readme

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#60 - Update README for Colocated Python Sidecar.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#59 - Group bulk write requests by device list. It avoids setting devices repeatedly for each request.

Pull Request - State: open - Opened by copybara-service[bot] 4 months ago - 1 comment

#58 - Adding an elastic manager and reshard modules

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#57 - Adding debugging utilities

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#56 - Skipping a broken unittest

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#55 - Add a test for proxy backend registration

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#54 - Upgrade minimum JAX version to 0.5.1

Pull Request - State: closed - Opened by shauryagup 4 months ago - 1 comment

#53 - Fix copybara rule

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#52 - Fixing copybara exclude rule

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#51 - Internal change

Pull Request - State: open - Opened by copybara-service[bot] 5 months ago

#50 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#49 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#48 - Use `| None` for `Optional` in type annotations and rename a module

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#47 - Introduce `pathwaysutils.initialize()` to remove relying on side-effects from `import pathwaysutils`.

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#46 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#45 - Separating the github workflows

Pull Request - State: closed - Opened by lukebaumann 6 months ago

#44 - Fixed GitHub Action

Pull Request - State: closed - Opened by lukebaumann 6 months ago

#43 - Updating GitHub actions

Pull Request - State: closed - Opened by lukebaumann 6 months ago

#42 - Updating publishing to PyPI actions

Pull Request - State: closed - Opened by lukebaumann 6 months ago

#41 - Updated the version to v0.0.8

Pull Request - State: closed - Opened by copybara-service[bot] 6 months ago

#40 - Fix Pathwaysutils tests to run with the JAX cpu platform.

Pull Request - State: closed - Opened by copybara-service[bot] 6 months ago

#39 - Updating the changelog in preparation for version v0.0.8

Pull Request - State: closed - Opened by lukebaumann 6 months ago

#38 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] 6 months ago - 1 comment

#37 - Drop xc symbol dependency for persistence reads.

Pull Request - State: closed - Opened by copybara-service[bot] 6 months ago - 1 comment

#36 - Change restore timeout to 1 hour

Pull Request - State: closed - Opened by copybara-service[bot] 6 months ago - 1 comment

#35 - internal change

Pull Request - State: closed - Opened by copybara-service[bot] 7 months ago - 1 comment

#34 - internal change

Pull Request - State: closed - Opened by copybara-service[bot] 7 months ago - 1 comment

#33 - update orbax handler to use bulk read APIs.

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago - 1 comment

#32 - Disable JAX's compilation cache since Pathways manages it's own compilation cache.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#31 - Uses the default python logging package instead of absl.logging

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#30 - Refactoring: move type_handlers.py to _src/serialization

Pull Request - State: open - Opened by copybara-service[bot] 9 months ago

#29 - Update main.yml

Pull Request - State: closed - Opened by lukebaumann 9 months ago

#28 - Bump version to v0.0.7 and trigger first PyPI release.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#27 - Preparing for v0.0.7 PyPI release

Pull Request - State: closed - Opened by lukebaumann 9 months ago

#26 - Add copyright and license headers to pathwaysutils test files.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#25 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#24 - Update pyproject.toml

Pull Request - State: closed - Opened by lukebaumann 9 months ago

#23 - Add a test directory with a few tests.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#22 - Create main.yml

Pull Request - State: closed - Opened by lukebaumann 9 months ago

#21 - Fixed a bug in the version format

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#20 - Preparing for releasing on PyPI.

Pull Request - State: closed - Opened by lukebaumann 10 months ago

#19 - Adding a package version variable.

Pull Request - State: closed - Opened by copybara-service[bot] 10 months ago

#18 - Update setup.py to prepare for v0.0.6 release

Pull Request - State: closed - Opened by lukebaumann 10 months ago

#17 - Logging updates

Pull Request - State: closed - Opened by copybara-service[bot] 10 months ago

#16 - Changed how `ifrt_proxy` is imported to improve backwards compatibility with JAX

Pull Request - State: closed - Opened by copybara-service[bot] 10 months ago

#15 - Fixed import in plugin_executable.py

Pull Request - State: closed - Opened by copybara-service[bot] 10 months ago

#14 - Avoid using relatively new `jax.extend` package until all modules that we use are part of a stable JAX release.

Pull Request - State: closed - Opened by copybara-service[bot] 10 months ago

#13 - Fixes name of proto field used for persistence and uses total_seconds to account for days.

Pull Request - State: closed - Opened by copybara-service[bot] 10 months ago

#12 - Change from using `jaxlib.xla_client.ifrt_programs` to `jax.extend.ifrt_programs.ifrt_programs`.

Pull Request - State: closed - Opened by copybara-service[bot] 10 months ago

#11 - Catch OSError when setting up cloud logging.

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#10 - Undoes the incorrect fix to the argument and instead fixes the type hints of the function for `write_executable.call`

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#9 - Internal Change Only

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago - 1 comment

#8 - Fixing an argument type so that the executable call is receiving the correct parameter type: `Sequence[List[jax.Array]]`

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#7 - Changing from deprecated `jax.sharding.XLACompatibleSharding` to `jax.sharding.Sharding` per deprecation warning: `jax.sharding.XLACompatibleSharding is deprecated. Use jax.sharding.Sharding instead.`

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#6 - Converting timeout for read and write requests to `int` to match the function signature types.

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#5 - Use `jax.lib.xla_client` instead of `jax._src.lib.xla_client` to use the public API.

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#4 - Use `jax.extend.backend` instead of `jax._src.xla_bridge` for registering the proxy backend.

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#3 - Re-enable Orbax persistence handler registration in pathwaysutils.

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago - 1 comment

#2 - Re-enable persistence behind an env var

Pull Request - State: closed - Opened by sadikneipp 11 months ago - 1 comment

#1 - Use `orbax-checkpoint` instead of `orbax`

Pull Request - State: closed - Opened by sadikneipp 11 months ago

GitHub / ai-hypercomputer/pathways-utils issues and pull requests