Code

Support KFAC for >2d model outputs#62

Code

Merged

runame merged 6 commits intodevelopmentf-dangel/curvlinops:developmentfrom high-dim-outputsf-dangel/curvlinops:high-dim-outputsCopy head branch name to clipboardJan 11, 2024

+146-40

Commits

Commits on Jan 9, 2024

• Add Rearrange module util

  runame

  committed
  c362dba

  Copy full SHA for c362dba

• Add test for KFAC with >2d model outputs

  runame

  committed
  5ae9553

  Copy full SHA for 5ae9553

• Add support for >2d linear layer inputs (expand) and model outputs

  runame

  committed
  2acd800

  Copy full SHA for 2acd800

• Fix black

  runame

  committed
  861ec7e

  Copy full SHA for 861ec7e

Commits on Jan 10, 2024

• Address review feedback

  runame

  committed
  33f7d50

  Copy full SHA for 33f7d50

Commits on Jan 11, 2024

• Remove superfluous if statement

  runame

  committed
  ad2dc3f

  Copy full SHA for ad2dc3f