Example on computing logprob with the model

[fountaindive]

Hi,

First of all this work is super impressive and thanks for doing it!

I asked about computing the density of a sample here. I was able to run this example notebook however, I couldn't figure out how to use the model in the following way

logprob = model.logprob(X)

For example, it would be really useful to be able evaluate the logprob of the model on any input sample if that makes sense!

This might be me not fully understanding the method but to evaluate the logprob do you have run the ODE solver for a sample going from the prior space to the target space? Or can you take a sample already in the target space and compute the logprob?

Thanks!