Learning to Transmit with Provable Guarantees in Wireless Federated Learning