search

jax-ml / coix

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

Add PyMC as a modeling language #34

Open twiecki opened 2 months ago

twiecki commented 2 months ago

PyMC, through PyTensor, can compile to JAX. It should thus be possible to add it here as a modeling language.

fehiepsi commented 2 months ago

Yes, it should be possible. That would be a great contribution!

We can introduce a coix.pymc module and implement the bridge transformations trace_evaluate, empirical there.