issues
search
google-deepmind
/
kfac-jax
Second Order Optimization and Curvature Estimation with K-FAC in JAX.
Apache License 2.0
235
stars
18
forks
source link
* Simplifying the LossTag machinery.
#240
Closed
copybara-service[bot]
closed
4 months ago
copybara-service[bot]
commented
4 months ago
Simplifying the LossTag machinery.
Fixing a few bugs introduced in the previous commit.