Some of the specialization comes from equipment - the support element packs extra machine guns and grenade launchers, while the assault element wants a more streamlined loadout.

It might be possible to task organize the units, but allow the men to rotate jobs once in a while by swapping out weapons when there's a quiet moment. Or would that cause too many complications?