The present disclosure relates to systems, methods, and non-transitory computer readable media for training and utilizing a generative machine learning model to select one or more treatments for a client device from a set of treatments based on digital characteristics corresponding to the client device. In particular, the disclosed systems can train and apply a variational autoencoder with a task embedding layer that generates estimated effects for treatment combinations. For example, the disclosed systems receive, as input, digital characteristics corresponding to the client device and various treatment combinations. The disclosed systems apply the trained generative machine learning model with the task embedding layer to the digital characteristics to generate effect estimations for the various treatment combinations. Based on the effect estimations for the treatment combinations, the disclosed systems select one or more treatments to provide to the client device.