Code for the IEEE-TSP Paper "Model-Free Learning of Two-Stage Beamformers for Passive IRS-Aided Network Design"