Skip to content

Pull requests: google-research/kauldron

New pull request New
46 Open 908 Closed
46 Open 908 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

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…
Replacate MultiTrainStep by the kd.contrib.train
#952 opened Nov 23, 2024 by copybara-service bot Loading…
Remove ModelWithAux in trainstep
#951 opened Nov 22, 2024 by copybara-service bot Loading…
Remove model_with_aux from write metrics
#950 opened Nov 22, 2024 by copybara-service bot Loading…
Remove deprecated arguments
#949 opened Nov 22, 2024 by copybara-service bot Loading…
Make module wrapper transparent with respect to the scope
#948 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…
ProTip! Mix and match filters to narrow down what you’re looking for.