feat: add projection_dim support to kronecker factored Hessians - #337

Merged

luciaquirke merged 3 commits into

mainfrom

feat/compressed-kfac-apply

Jul 19, 2026

Commits on Jul 18, 2026

feat: compress K-FAC IVHP output to match compressed gradient stores

committed
Reject unsupported EK-FAC-compression and cosine-sim-with-factored-Hessian combos

committed
Clarify bergson score's error for factored hessian + projection + unit_normalize

committed