Xtr training perf - #2

Merged

robro612 merged 2 commits into

robro612:xtr-trainingfrom

arn4:xtr-training-perf

Apr 4, 2026

Commits on Apr 2, 2026

Decoupled mini_batch_size and score_mini_batch_size for effective GradCache without OOM during score computation.
arn4
committed

Commits on Apr 4, 2026

Speed up implementation of XTRScore. Implemented the compilation.
arn4
committed