-
Zimin Li authored
issue/261: optimize the torch implementation of add, causal softmax, gemm, random sample, rearrange, rms norm, rope
505e0d4b
issue/261: optimize the torch implementation of add, causal softmax, gemm, random sample, rearrange, rms norm, rope