Introduce the `chain` argument to propose

jax-ml / coix

Inference Combinators in JAX

https://coix.readthedocs.io/en/latest/

Apache License 2.0

43 stars 2 forks source link

Introduce the `chain` argument to propose #29

Closed fehiepsi closed 6 months ago

fehiepsi commented 7 months ago

When proposing q to p, sometimes we want to pass information from q into p. This PR introduces a chain argument to optionally use output of q as input of p.