Various aspects of the present disclosure generally relate to wireless communication. In some aspects, a base station may receive, from a user equipment (UE), a measurement report associated with at least one downlink beam corresponding to a first transmit receive point (TRP). The base station may transmit, to the UE and using a second TRP, a communication using a selected downlink beam corresponding to the second TRP, wherein the selected downlink beam is selected based at least in part on a beam inference generated using a machine learning component, and wherein the beam inference is based at least in part on the measurement report and a location of the second TRP relative to a location of the first TRP. Numerous other aspects are described.