Making large AI models cheaper, faster and more accessible
COMMITS
/ examples/inference/benchmark_ops/benchmark_rmsnorm.py April 19, 2024
S
feat baichuan2 rmsnorm whose hidden size equals to 5120 (#5611)
Steve Luo committed
March 8, 2024
S
feat rmsnorm cuda kernel and add unittest, benchmark script (#5417)
Steve Luo committed