jax-ml / coix

Inference Combinators in JAX
https://coix.readthedocs.io/en/latest/
Apache License 2.0
43 stars 2 forks source link

Tutorial for annealed flow transport #10

Open fehiepsi opened 1 year ago

fehiepsi commented 1 year ago

Annealed Flow Transport Monte Carlo, Arbel etc. can be implemented using a few combinators as in coix.algo. A special aspect of this algorithm is its use of "deterministic" kernels rather than "stochastic" kernels as in inference combinators.

It would be great if we have a concrete example/tutorial showing how to implement targets and flows. Here, the role of flows is to perform a transformation and add a corresponding log_det_abs_jacobian factor.

conorhassan commented 5 months ago

hi there, would like to try and give this a try

fehiepsi commented 5 months ago

That would be awesome, @conorhassan!