fix: torch.compile loss corruption at ignore_index rows (+ perf: drop BT x V loss buffer in FLJSD) - #1338

fix: torch.compile loss corruption at ignore_index rows (+ perf: drop BT x V loss buffer in FLJSD)#1338

KyleMylonakisProtopia wants to merge 5 commits into

linkedin:mainlinkedin/Liger-Kernel:mainfrom

KyleMylonakisProtopia:fix/torch-compile-ignore-index-loss-corruptionKyleMylonakisProtopia/Liger-Kernel:fix/torch-compile-ignore-index-loss-corruptionCopy head branch name to clipboard