GreatDrake/jem-jax

jax implementation of joint energy-based model

Python

jem-jax

JAX implementation of Joint Energy-based Model from Your Classifier is Secretly an Energy Based Model and You Should Treat it Like One.