Skip to content

Do the chunked-CE lm_head projection on tensor cores instead of in fp32 #3753

Do the chunked-CE lm_head projection on tensor cores instead of in fp32

Do the chunked-CE lm_head projection on tensor cores instead of in fp32 #3753