The present disclosure describes a wafer cooling/heating system that includes a load-lock and a thermo module. The load-lock uses a level stream design to improve temperature uniformity across one or more wafers during a cooling/heating process. The load-lock can include (i) a wafer holder configured to receive wafers at a front side of the load-lock; (ii) a gas diffuser with one or more nozzles along a back side of the load-lock, a side surface of the load-lock, or a combination thereof; and (iii) one or more exhaust lines. Further, the thermo module can be configured to control a temperature of a gas provided to the load-lock.