feat: Arbitrary-Rank Ablation (ARA)#211

Draft

p-e-w wants to merge 17 commits intomasterfrom

Commits on Feb 25, 2026

feat(ara): add methods for obtaining module I/O
p-e-w
committed

Commits on Feb 27, 2026

feat(ara): implement matrix optimization
p-e-w
committed

Commits on Mar 2, 2026

Commits on Mar 5, 2026

feat(ara): optimize all parameters
p-e-w
committed

Commits on Mar 7, 2026

feat(ara): fix issues on some multi-GPU setups

p-e-w
and
kabachuha
committed
fix(ara): store captured I/O tensors on CPU for multi-GPU robustness (#214)
joninco
authored

Commits on Mar 10, 2026

Commits on Mar 21, 2026

feat(ara): add option to optimize for PIQA instead of KLD
p-e-w
committed

Commits on Mar 30, 2026

Merge branch 'master' into ara
p-e-w
committed

Commits on Mar 31, 2026

feat(ara): add optional row-norm preservation
p-e-w
committed

Commits on Apr 1, 2026

fix(ara): set batch size on HFLM object
p-e-w
committed