Skip to content

Pull requests: google/orbax

New pull request New
83 Open 808 Closed
83 Open 808 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
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

Add logging for emergency checkpoint manager.
#981 opened Jun 29, 2024 by copybara-service bot Loading…
remove name arg of JaxModule which never use.
#979 opened Jun 27, 2024 by copybara-service bot Loading…
Create the orbax.export typing API
#978 opened Jun 27, 2024 by copybara-service bot Loading…
Make atomicity logic configurable, and encapsulate it within a class.
#974 opened Jun 26, 2024 by copybara-service bot Loading…
Introduce replica_axis_index in init of emergency checkpoint manager
#968 opened Jun 21, 2024 by copybara-service bot Loading…
1
test copybara, don't submit
#965 opened Jun 20, 2024 by copybara-service bot Loading…
Roll forward: Improve tensorstore I/O efficiency
#961 opened Jun 20, 2024 by copybara-service bot Loading…
1
Add memory limiting functionality for save, in addition to restore.
#956 opened Jun 17, 2024 by copybara-service bot Loading…
Orbax: Refactor tensorstore spec construction and support local temporary index path
#945 opened Jun 11, 2024 by copybara-service bot Loading…
1
[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…
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.