Skip to content

Pull requests: google-research/t5x

New pull request New
100 Open 1,273 Closed
100 Open 1,273 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

depend on latest orbax-checkpoint from github instead of pypi
#1563 opened May 22, 2024 by copybara-service bot Loading…
Set aggregate to False under SaveArgs within t5x.
#1561 opened May 17, 2024 by copybara-service bot Loading…
Enable ocdbt in t5x by default.
#1560 opened May 17, 2024 by copybara-service bot Loading…
End support for legacy Pax formats with directory structure:
#1558 opened May 16, 2024 by copybara-service bot Loading…
Delegate to BasePyTreeCheckpointHandler rather than inheriting from it.
#1556 opened May 15, 2024 by copybara-service bot Loading…
Added MAXO-GEN models, namely, SAC, AGIQA-1k, AGIQA-3k and AIGCIQA-2023.
#1543 opened Apr 19, 2024 by copybara-service bot Loading…
1
Added MAXO-GEN models, namely, SAC, AGIQA-1k, AGIQA-3k and AIGCIQA-2023.
#1542 opened Apr 19, 2024 by copybara-service bot Loading…
1
Rename dpsgd to orbax.contrib END_PUBLIC
#1540 opened Apr 12, 2024 by copybara-service bot Loading…
Rename DifferentiallyPrivateAggregateState to orbax.contrib
#1539 opened Apr 12, 2024 by copybara-service bot Loading…
jax.tree.map->jax.tree_map
#1535 opened Mar 28, 2024 by jntdst Loading…
1
DNS test cloud issue.
#1531 opened Mar 20, 2024 by copybara-service bot Loading…
fix eval
#1522 opened Mar 4, 2024 by copybara-service bot Loading…
internal
#1469 opened Dec 17, 2023 by copybara-service bot Loading…
1
internal
#1468 opened Dec 17, 2023 by copybara-service bot Loading…
1
internal
#1467 opened Dec 17, 2023 by copybara-service bot Loading…
1
internal
#1466 opened Dec 17, 2023 by copybara-service bot Loading…
1
Remove tensorflow-cpu dependency
#1464 opened Dec 15, 2023 by vedansh0739 Loading…
1
Replace references to deprecated device_buffer attributes
#1447 opened Dec 6, 2023 by copybara-service bot Loading…
Unpins tfds-nightly in gpu extra which is in conflict with seqio's specification
#1416 opened Oct 11, 2023 by terrykong Loading…
Filter INF values from weight_update_divided_by_weight_rms.
#1404 opened Sep 21, 2023 by copybara-service bot Loading…
1
Add a jax context manager to prevent crashes in some cases.
#1402 opened Sep 20, 2023 by copybara-service bot Loading…
1
Add DALI support to t5x
#1393 opened Sep 8, 2023 by terrykong Loading…
1
Add support for partial checkpoint restore
#1392 opened Sep 8, 2023 by terrykong Loading…
1
[NVIDIA] Added transformer engine support and GPU optimizations
#1391 opened Sep 8, 2023 by terrykong Loading…
1
Fix broken type annotations, and replace pytype disables with typing.cast
#1384 opened Aug 25, 2023 by copybara-service bot Loading…
1
ProTip! What’s not been updated in a month: updated:<2024-04-22.