A method implemented in a computer system includes training a network, which is obtained by unfolding an iterative algorithm for demodulation or demodulation and decoding, using a machine learning technique with a loss function that takes into account non-Gaussianity of a log-likelihood ratio (LLR) distribution calculated from an output of the network. The method further includes producing a first set of learned parameters of that iterative algorithm.