Imported from https://github.com/conormdurkan/autoregressive-energy-machines