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

Split `curvature_estimator.py` module into a package. #261

Closed copybara-service[bot] closed 1 month ago

copybara-service[bot] commented 1 month ago

Split curvature_estimator.py module into a package.