Fix RMS norm patching

[vaibhavjindal]

Summary

Fixes #739.

Creates new classes for various RMS norms and removes the use of partial for RMS norms.

Testing Done

Fixes errors of the form:

TypeError: isinstance() arg 2 must be a type, a tuple of types, or a union

for glm4, olmo2, gemma1 and gemma2.

However, we are now seeing errors when matching the actual logits with gemma models, which can be tracked separately in #729.

• Hardware Type:
• run make test to ensure correctness
• run make checkstyle to ensure code style
• run make test-convergence to ensure convergence

[Tcc0403]

Thanks!