pytorch-jem Your classifier is secretly an energy based model and you should treat it like one https://arxiv.org/abs/1912.03263