Edit: nevermind, found it. For those looking, it's at https://github.com/harvardnlp/NeuralSteganography