Do the chunked-CE lm_head projection on tensor cores instead of in fp32
#3753
| Job | Run time |
|---|---|
| 5m 58s | |
| 13s | |
| 6m 11s |
lm_head projection on tensor cores instead of in fp32
#3753
| Job | Run time |
|---|---|
| 5m 58s | |
| 13s | |
| 6m 11s |