fix: AuxK loss carries no gradient, now backprop combined loss, not reconstruction only - #3

fix: AuxK loss carries no gradient, now backprop combined loss, not reconstruction only#3

SohrabTa wants to merge 1 commit into

oclivegriffin:mainoclivegriffin/crosscode:mainfrom

SohrabTa:fix/auxk-loss-carries-no-gradientSohrabTa/crosscode:fix/auxk-loss-carries-no-gradientCopy head branch name to clipboard