Skip to content

Pull requests: google-research/kauldron

New pull request New
44 Open 918 Closed
44 Open 918 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 kd.train.Auxiliaries -> kd.train.AuxiliariesState
#958 opened Nov 27, 2024 by copybara-service bot Loading…
Add return_context to trainstep.step()
#955 opened Nov 25, 2024 by copybara-service bot Loading…
Have ._step returns the full Context
#954 opened Nov 23, 2024 by copybara-service bot Loading…
Use public aliases to define MultiTrainStep
#953 opened Nov 23, 2024 by copybara-service bot Loading…
Remove ModelWithAux in trainstep
#951 opened Nov 22, 2024 by copybara-service bot Loading…
Update data pipeline to support post batch transforms
#939 opened Nov 14, 2024 by copybara-service bot Loading…
1
Make all data imports lazy
#920 opened Oct 31, 2024 by copybara-service bot Loading…
Adding NormalizeToRange op and support for unqiue_labels in the RocAuc computation.
#914 opened Oct 29, 2024 by copybara-service bot Loading…
Replace prefetch with mp_prefetch in PyGrain.
#912 opened Oct 25, 2024 by copybara-service bot Loading…
Convert shape spec parsing to a standalone parser to remove the lark dependency
#868 opened Sep 23, 2024 by copybara-service bot Loading…
Code update
#863 opened Sep 19, 2024 by copybara-service bot Loading…
Fix flags reload on colab
#855 opened Sep 11, 2024 by copybara-service bot Loading…
WIP: fix the state.parent for TreeReduce
#847 opened Sep 9, 2024 by copybara-service bot Loading…
Do not re-randomize when shuffle=False
#822 opened Aug 27, 2024 by copybara-service bot Loading…
Add the option to process params with a custom function before writing them.
#803 opened Aug 16, 2024 by copybara-service bot Loading…
1
Implement optional transforms before cache.
#797 opened Aug 5, 2024 by copybara-service bot Loading…
File saving to track end of evaluation runs. This helps with extensions that might wait for some evals to finish before running an op.
#781 opened Jul 10, 2024 by copybara-service bot Loading…
Support launching jobs in an existing xmanager experiment.
#687 opened May 22, 2024 by copybara-service bot Loading…
1
Add 2D RoPE embedding to Kauldron.
#683 opened May 20, 2024 by copybara-service bot Loading…
1
Fix data/utils dtype warning for float64 inputs.
#675 opened May 9, 2024 by copybara-service bot Loading…
1
Serialize sets for Json.
#670 opened Apr 29, 2024 by copybara-service bot Loading…
1
Support setting num_epochs instead of num_training_steps.
#652 opened Apr 17, 2024 by copybara-service bot Loading…
1
Make g3pdb lazy again
#649 opened Apr 15, 2024 by copybara-service bot Loading…
Raise an error when using TFDataPipeline
#589 opened Mar 19, 2024 by copybara-service bot Loading…
modify ModelWithAux for use with custom train steps and multi-task learning.
#536 opened Mar 4, 2024 by copybara-service bot Loading…
ProTip! Filter pull requests by the default branch with base:main.