DeepSpeed

a53f298e - change matmul with F.linear (seems to have better performance)