- Notifications
You must be signed in to change notification settings - Fork 28
Pull requests: google-deepmind/kfac-jax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
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…
Only check adaptive damping settings if adaptive learning rate is enabled.
#320 opened Mar 6, 2025 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…
ProTip! What’s not been updated in a month: updated:<2025-10-17.