Skip to content

Pull requests: google/init2winit

New pull request New
58 Open 712 Closed
58 Open 712 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
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Some refactoring of methods to enable additional customization of BaseTrainer, including how input data to models are sharded.
#770 opened Jun 3, 2025 by copybara-service bot Loading…
Internal
#766 opened May 15, 2025 by copybara-service bot Loading…
Internal
#760 opened May 5, 2025 by copybara-service bot Loading…
Internal change.
#740 opened Apr 9, 2025 by copybara-service bot Loading…
1
Internal change
#737 opened Apr 1, 2025 by copybara-service bot Loading…
Internal change.
#730 opened Mar 28, 2025 by copybara-service bot Loading…
internal change
#716 opened Mar 1, 2025 by copybara-service bot Loading…
1
Internal
#705 opened Jan 30, 2025 by copybara-service bot Loading…
Update references to JAX's GitHub repo
#690 opened Dec 5, 2024 by copybara-service bot Loading…
Bump tensorflow from 2.5.0 to 2.12.1 in the pip group dependencies Pull requests that update a dependency file
#677 opened Jul 30, 2024 by dependabot bot Loading…
Stop writing msgpack file for new checkpoints and update empty nodes handling so that it no longer depends on this file.
#673 opened Jun 27, 2024 by copybara-service bot Loading…
Adding schedule free with adam and nadam
#672 opened Jun 12, 2024 by copybara-service bot Loading…
1
Internal.
#670 opened May 14, 2024 by copybara-service bot Loading…
Proposal to make 'eval_use_ema' a hyper-parameter visible to optimizers.
#669 opened May 14, 2024 by copybara-service bot Loading…
Adding the meta-optimization trainer to i2w.
#656 opened Apr 16, 2024 by copybara-service bot Loading…
1
Internal
#636 opened Mar 20, 2024 by copybara-service bot Loading…
Add PAL optimizer.
#626 opened Feb 16, 2024 by copybara-service bot Loading…
FKM dynamic sketchy memory reallocation
#610 opened Dec 29, 2023 by copybara-service bot Loading…
1
Internal
#590 opened Sep 27, 2023 by copybara-service bot Loading…
[flax] Remove redundant deterministic parameter from dot_product_attention.
#588 opened Sep 25, 2023 by copybara-service bot Loading…
Expts
#583 opened Aug 25, 2023 by copybara-service bot Loading…
1
Experiments
#582 opened Aug 25, 2023 by copybara-service bot Loading…
1
Small fix to the model.params_types rules.
#567 opened Jul 20, 2023 by copybara-service bot Loading…
internal
#557 opened Jun 12, 2023 by copybara-service bot Loading…
Add matrix epsilon diagonal instead of truncating numerical noise.
#555 opened Jun 11, 2023 by copybara-service bot Loading…
ProTip! Add no:assignee to see everything that’s not assigned.