* perf(pi052): optimize equivalent training paths * fix(pi052): guard FlexAttention backend selection