Skip to content

Pull requests: google/orbax

New pull request New
70 Open 762 Closed
70 Open 762 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Internal change.
#917 opened May 31, 2024 by copybara-service bot Loading…
Internal change.
#916 opened May 31, 2024 by copybara-service bot Loading…
Internal change.
#915 opened May 30, 2024 by copybara-service bot Loading…
Swap jax.process_index() for jax._src.distributed.global_state.process_id.
#914 opened May 30, 2024 by copybara-service bot Loading…
Internal change.
#913 opened May 30, 2024 by copybara-service bot Loading…
Refactor functions from utils to tree package.
#911 opened May 29, 2024 by copybara-service bot Loading…
Stop writing msgpack file for new checkpoints and update empty nodes handling so that it no longer depends on this file.
#908 opened May 28, 2024 by copybara-service bot Loading…
Fix a typo in exception message.
#905 opened May 28, 2024 by copybara-service bot Loading…
1
Internal change.
#904 opened May 28, 2024 by copybara-service bot Loading…
Internal
#902 opened May 26, 2024 by copybara-service bot Loading…
End support for legacy Pax formats with directory structure:
#882 opened May 16, 2024 by copybara-service bot Loading…
Fork a small amount of Orbax code into Pax to deal with writing "aggregate" files, as Orbax will soon lose this ability.
#869 opened May 8, 2024 by copybara-service bot Loading…
Add try catch block to catch ValueError reading from sharding string.
#857 opened May 1, 2024 by copybara-service bot Loading…
Add OCDBT support to GlobalHostArray type_handler
#843 opened Apr 24, 2024 by copybara-service bot Loading…
Add an option to allow sharding consolidation / reduction to work around the dtensor limitation of disallowing one dimension being sharded across multiple axis names.
#822 opened Apr 15, 2024 by copybara-service bot Loading…
1
Create the utils function load_model_with_tfrt API for XLA devices.
#813 opened Apr 12, 2024 by copybara-service bot Loading…
Internal change.
#801 opened Apr 6, 2024 by copybara-service bot Loading…
Internal change.
#800 opened Apr 6, 2024 by copybara-service bot Loading…
Enable Orbax checkpointing with background deleting thread
#796 opened Apr 4, 2024 by copybara-service bot Loading…
remove all_params_aggregated.
#794 opened Apr 3, 2024 by copybara-service bot Loading…
Internal
#783 opened Apr 2, 2024 by copybara-service bot Loading…
1
Create an interface for retrieving the current step number.
#778 opened Mar 30, 2024 by copybara-service bot Loading…
1
Internal change.
#754 opened Mar 15, 2024 by copybara-service bot Loading…
file access hardening on model exporter.
#717 opened Feb 14, 2024 by copybara-service bot Loading…
1
Changes default timeout in AsyncCheckpointer.
#698 opened Feb 1, 2024 by copybara-service bot Loading…
ProTip! Filter pull requests by the default branch with base:main.