Various aspects of the present disclosure generally relate to wireless communication. In some aspects, a user equipment (UE) may receive a transformer configuration that includes a transmitter neural network configured to be used to generate at least one latent vector corresponding to one or more computation tasks of a plurality of computation tasks associated with a transformer-based cross-node machine learning system. The UE may transmit the at least one latent vector based at least in part on instantiating the transmitter neural network. Numerous other aspects are described.