feat(inference): fp32 lm_head via native bf16xbf16 -> fp32 mm (alt to #2438)#2441

samsja merged 1 commit into

mainPrimeIntellect-ai/prime-rl:mainfrom

feat/fp32-lm-head-out-dtypePrimeIntellect-ai/prime-rl:feat/fp32-lm-head-out-dtypeCopy head branch name to clipboard