Hutchinson Trace Estimator Version

Hello,

First of all, thank you very much for such a great simplification of FFJORD code.

Would it be possible for you to also add a high-dimension version that uses HTE?

Sincerely,
Kamer