probtorch / pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Home Page:http://pytorch.org

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Implement Concrete distribution

alicanb opened this issue · comments

Duplicate of #2. See example implementation by @rachtsingh at pyro-ppl/pyro#588

Sorry, my bad

It's nice that we're getting so much done we lose track of what remains 😉

Ah yeah, let's port this after Bijectors land? I think I still need to write some tests, but the failures were PyTorch 0.3 stuff. Just looked over it, I was refactoring some of the Jacobian check code so that it's generic, and then using that both in the relaxed distributions and IAF.