Skip to content

Pull requests: google-deepmind/kfac-jax

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

Silence some pytype errors related to a JAX build refactor
#349 opened Jun 16, 2025 by copybara-service bot Loading…
Some refactoring of methods to enable additional customization of BaseTrainer, including how input data to models are sharded.
#345 opened Jun 3, 2025 by copybara-service bot Loading…
[JAX] Fix Jaxpr comparisons for upcoming change to DropVar behavior.
#332 opened Apr 23, 2025 by copybara-service bot Loading…
jex.core.Var is no longer ordered
#328 opened Apr 7, 2025 by copybara-service bot Loading…
Only check adaptive damping settings if adaptive learning rate is enabled.
#320 opened Mar 6, 2025 by copybara-service bot Loading…
Update references to JAX's GitHub repo
#306 opened Dec 5, 2024 by copybara-service bot Loading…
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! Follow long discussions with comments:>50.