Skip to content

Pull requests: google-research/t5x

New pull request New
95 Open 1,326 Closed
95 Open 1,326 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

Rename best_fn to get_metric_fn in CheckpointManagerOptions.
#1617 opened Apr 23, 2025 by copybara-service bot Loading…
Add cast to fix conversion to int64 when JAX's x64 mode is on.
#1615 opened Apr 11, 2025 by copybara-service bot Loading…
1
Fix a bug in Orbax checkpointing where None values PyTree are not handled correctly. This is caused by a recent update in jax where None values are no longer considered as a leaf node: https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-34-october-4-2024.
#1604 opened Oct 31, 2024 by copybara-service bot Loading…
1
Disable golden tests in contributed gpu layers_test.py
#1600 opened Oct 21, 2024 by levskaya Loading…
add wait_until_finished under checkpoint_manager to fix issue when checkpoint was not completed.
#1581 opened Jul 17, 2024 by copybara-service bot Loading…
Sort devices by their implicit order instead of explicitly by id. IDs may be randomly generated, so it's better to rely on the implicit order, which is currently based on (process index, id).
#1578 opened Jul 10, 2024 by copybara-service bot Loading…
1
Revert get_local_data to original implementation.
#1571 opened Jun 12, 2024 by copybara-service bot Loading…
Fix bug in T5X CheckpointManager argument propagation when SaveBestCheckpoint is set through gin.
#1568 opened Jun 7, 2024 by copybara-service bot Loading…
1
Propagate metric_name_to_monitor to Orbax CheckpointManager.
#1566 opened Jun 3, 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
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 VedB0739 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…
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
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.