Skip to content

Pull requests: google/orbax

New pull request New
79 Open 779 Closed
79 Open 779 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

Orbax: Refactor tensorstore spec construction and support local temporary index path
#945 opened Jun 11, 2024 by copybara-service bot Loading…
1
Internal change.
#944 opened Jun 11, 2024 by copybara-service bot Loading…
Make broadcasting from one replica to all more memory efficient
#943 opened Jun 11, 2024 by copybara-service bot Loading…
1
Internal change.
#942 opened Jun 11, 2024 by copybara-service bot Loading…
[JAX] Update users of jax.tree.map() to be more careful about how they handle Nones.
#938 opened Jun 9, 2024 by copybara-service bot Loading…
Internal change.
#935 opened Jun 7, 2024 by copybara-service bot Loading…
Internal change.
#931 opened Jun 5, 2024 by copybara-service bot Loading…
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…
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…
ProTip! Exclude everything labeled bug with -label:bug.