Add vanilla shampoo - #287

Open

skyw wants to merge 16 commits into

mainfrom

skyw/vanilla_shampoo_dev

Commits on Aug 20, 2026

Use type error for shape mismatch
skyw
committed
41a4879
add vanilla shampoo
skyw
committed
fe48cae
introduce base class for shampoo
skyw
committed
4ae48bd
save work
skyw
committed
72321f9
add test for vanilla shampoo
skyw
committed
5462eb4
rename module
skyw
committed
6993ab4
fix error type
skyw
committed
3e38538
improve test coverage
skyw
committed
20b78d0
update import order
skyw
committed
edb7040
add tikhonov and bias correction
skyw
committed
91d5c09
fix tests
skyw
committed
5e2ed4e
add test for shampoo beta bias correction
skyw
committed
b6140c5
add test for tikhonov
skyw
committed
0a504e6
update tests
skyw
committed
4b41c61
renamve to p_root_inv
skyw
committed
f7c6e0f
rename invere root to root inverse
skyw
committed
15fa3a1