-
Notifications
You must be signed in to change notification settings - Fork 305
Pull requests: google-research/t5x
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Move
metadata
package to _src
. Continue exposing commonly-used symbols.
#1602
opened Oct 24, 2024 by
copybara-service
bot
Loading…
Transition more of t5x export to jax2tf native serialization
#1601
opened Oct 22, 2024 by
copybara-service
bot
Loading…
Disable golden tests in contributed gpu layers_test.py
#1600
opened Oct 21, 2024 by
levskaya
Loading…
De-duplicate
get_ts_context
usages and move to ts_utils.
#1599
opened Oct 16, 2024 by
copybara-service
bot
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…
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…
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…
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…
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…
Previous Next
ProTip!
no:milestone will show everything without a milestone.