Add TensorFlow Probability as the modeling language

jax-ml / coix

Inference Combinators in JAX

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

Apache License 2.0

43 stars 2 forks source link

Open fehiepsi opened 6 months ago

fehiepsi commented 6 months ago

We can mimic coix.numpyro to implement traced_evaluate and empirical for a coix.tfp module.