[PyTorch] Enable single grouped tensor weight in GroupedLinear

[ptrendx]

Allow GroupedLinear layers in PyTorch to accept a single grouped tensor weight input. Refactor and extend the interface to use grouped weights efficiently during forward and backward passes.

[ksivaman]

Closed in #2731 and #2736