Gallium nitride (GaN) power stages