-
Notifications
You must be signed in to change notification settings - Fork 293
Pull requests: google-research/t5x
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Set
aggregate
to False under SaveArgs within t5x.
#1561
opened May 17, 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…
Added MAXO-GEN models, namely, SAC, AGIQA-1k, AGIQA-3k and AIGCIQA-2023.
#1542
opened Apr 19, 2024 by
copybara-service
bot
Loading…
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…
Add a jax context manager to prevent crashes in some cases.
#1402
opened Sep 20, 2023 by
copybara-service
bot
Loading…
[NVIDIA] Added transformer engine support and GPU optimizations
#1391
opened Sep 8, 2023 by
terrykong
Loading…
Fix broken type annotations, and replace pytype disables with typing.cast
#1384
opened Aug 25, 2023 by
copybara-service
bot
Loading…
Updates train.py to support initialization_timeout in jax.distributed.initialize
#1365
opened Aug 14, 2023 by
terrykong
Loading…
[NVIDIA] Added transformer engine support and GPU optimizations
#1320
opened Jun 27, 2023 by
SahilJain314
Loading…
Ensure that topp_mask keeps the largest logit unmasked.
#1304
opened Jun 8, 2023 by
copybara-service
bot
Loading…
Previous Next
ProTip!
Filter pull requests by the default branch with base:main.