it has been done here: https://arxiv.org/abs/1712.02734