Skip to content

Pull requests: google-deepmind/kfac-jax

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

Remove invalid type check.
#370 opened Dec 15, 2025 by copybara-service bot Loading…
Upgrade GitHub Actions to v6 for Node 24 compatibility
#369 opened Dec 13, 2025 by salmanmkc Loading…
- Changing all methods/attributes of the optimizer classes that should be private (which is all that are not part of the public-facing interface to the classes) to be private.
#368 opened Dec 10, 2025 by copybara-service bot Loading…
Internal Change
#358 opened Sep 11, 2025 by copybara-service bot Loading…
1
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! What’s not been updated in a month: updated:<2025-11-15.