Kai Xu

OK it works on GPU now. I did the following changes - fix a bug in sending data to GPU - remove broadcast ops in `logp_x_z` and `kl_q_p` - I...

I tried that naively but it doesn't successfully compile

What I am trying to do is to allow DIstributions.jl (and its dependency functions) to support AD by AutoGrad. - Most of the functions in DIstributions.jl allow `Real` to pass...

@denizyuret If my understanding is correct, as long as the underlying code is written in pure Julia and the corresponding primitive functions is extended to handle Rec, I can use...

Yes I understand this. The problem is not about the primitives but the function signature Distributions.jl and its dependencies use. Let me put it in an example. AutoGrad.jl works in...

Well you can do it but the point here is that `p(x)` is usually a very complex function, e.g. `logpdf` of a distribution. We really don't want to derive the...

I tried to simply add `

Also will you consider to add a compilation flag to let AutoGrad.jl set `Rec

Any idea on this?

Sounds like using Distributions.jl together with Bijectors.jl could work.