GreatDrake / jem-jax

jax implementation of joint energy-based model

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.