An apparatus for controlling wafer uniformity is disclosed. In one example, the apparatus includes: a plurality of temperature control elements and a processor. Each of the temperature control elements corresponds to a different portion of a wafer respectively such that the temperature control elements correspond to different portions of the wafer. Each of the temperature control elements is configured to individually control temperature of a corresponding portion of the wafer. The processor determines at least one portion of the wafer for temperature uniformity control, and instruct at least one of the temperature control elements, corresponding to the at least one portion, to adjust temperature of the at least one portion for controlling temperature uniformity of the wafer.