Skip to content

Pull requests: google/orbax

New pull request New
77 Open 771 Closed
77 Open 771 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

Enable standard logger by default and Fix synchronous flag input data in logging.
#934 opened Jun 6, 2024 by copybara-service bot Loading…
Add checkpoint format docs page.
#933 opened Jun 6, 2024 by copybara-service bot Loading…
Internal change.
#931 opened Jun 5, 2024 by copybara-service bot Loading…
Improve tensorstore I/O efficiency
#929 opened Jun 5, 2024 by copybara-service bot Loading…
1
Internal change.
#928 opened Jun 4, 2024 by cpgaffney1 Loading…
Internal change.
#926 opened Jun 4, 2024 by copybara-service bot Loading…
Internal change.
#925 opened Jun 4, 2024 by copybara-service bot Loading…
Internal change.
#923 opened Jun 4, 2024 by copybara-service bot Loading…
*** DNS *** Demo code to fix eval job killed due to low duty cycle.
#922 opened Jun 4, 2024 by copybara-service bot Loading…
1
Internal change.
#918 opened May 31, 2024 by copybara-service bot Loading…
Internal change.
#917 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…
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…
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…
ProTip! no:milestone will show everything without a milestone.