Skip to content

Pull requests: google-deepmind/kfac-jax

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

Adding support for models that contain conds, as long as the branches don't register any losses or directly use model parameters.
#308 opened Jan 7, 2025 by copybara-service bot Loading…
Update references to JAX's GitHub repo
#306 opened Dec 5, 2024 by copybara-service bot Loading…
chore(build): drop MANIFEST.in
#298 opened Nov 16, 2024 by SauravMaheshkar Loading…
4
Stackless yashful
#272 opened Oct 3, 2024 by copybara-service bot Loading…
Bump numpy from 1.21 to 1.22.0 dependencies Pull requests that update a dependency file
#165 opened Sep 4, 2023 by dependabot bot Loading…
Bump ipython from 7.16.1 to 8.10.0 dependencies Pull requests that update a dependency file
#164 opened Sep 4, 2023 by dependabot bot Loading…
Making KFAC use pjit instead of pmap
#121 opened May 2, 2023 by copybara-service bot Loading…
ProTip! no:milestone will show everything without a milestone.