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

Add more checks around compute quad model #228

Closed copybara-service[bot] closed 4 months ago

copybara-service[bot] commented 4 months ago

Add more checks around compute quad model