kfac-jax icon indicating copy to clipboard operation
kfac-jax copied to clipboard

Published 20 hours ago • verified publisher icon google-deepmind

* Simplifying type annotations in curvature_blocks.py

Open copybara-service[bot] opened this issue 1 year ago • 0 comments

  • Simplifying type annotations in curvature_blocks.py
  • Adding more general functions for Kronecker products multiplications.

copybara-service[bot] avatar Oct 07 '22 14:10 copybara-service[bot]