Expose grouped tensor type internally in PyTorch modules

[ptrendx]

Modify relevant PyTorch modules to internally use and support the GroupedTensor type, ensuring transparent integration for grouped MoE operations.

[ksivaman]

Closed by #2600, #2654, and #2731