Methods, systems, and apparatus, including computer programs encoded on computer storage media, for optical communications. One optical amplifier includes an input port; a bar-cross switch optically coupled to the input port; a first gain stage optically coupled between first port of the bar-cross switch and to an output port; and a secondary gain stage optically coupled between a second port and a third port of the bar-cross switch, wherein in a bar-state of the bar-cross switch the secondary gain stage is bypassed and in a cross-state, the secondary gain stage and the first gain stage are applied to an input light beam.