fix(rms_norm.py): add assertion for input gradients to handle unsupported cases in backward pass

feat: support sequence parallel with fused_add_rms_norm

refactor(activation): change fused_add_rms_norm and fused_add_rms_norm_backward to out-place operations

refactor(rms_norm): move RMS normalization logic to a new module for better organization and maintainability

feat: support sequence parallel with rms_norm

feat: make rms_norm as out-place

Fix fused add rms norm (#4)

Add fusion (#3)

Optimize kernel (#2)

feat: support reset_parameters()

feat(rms-norm): Impl fused RMSNorm

refactor(poly-norm): use const for immutable args

chore: use latest build image and misc

feat(poly-norm): add default value for eps argument

feat(poly-norm): Add PolyNorm