A method of wireless communication by a user equipment (UE) includes receiving, from a network device, a request to initiate gradient computations for a round of federated learning. The method also includes computing gradients in response to receiving the request to initiate gradient computation. The method further includes informing the network device of availability of the gradients. The method still further includes receiving, from the network device, information to enable transfer of the gradients to the network device. The method further includes transferring the gradients to the network device in response to receiving the information to enable transfer.