search

BasisResearch / chirho

An experimental language for causal reasoning
https://basisresearch.github.io/chirho/getting_started.html
Apache License 2.0
168 stars 11 forks source link

Support NumPyro and JAX #31

Open eb8680 opened 1 year ago

eb8680 commented 1 year ago

The key features in this library are not specific to Pyro or PyTorch. Can we make their implementations compatible with NumPyro and its JAX backend without duplicating lots of code?

qinqian commented 1 year ago

This looks interesting and a good opportunity to learn both probabilistic framework, let me know if I am capable to contribute this feature.

eb8680 commented 1 year ago

@qinqian thanks for volunteering! This is a bit of a longer-term goal, since it's not yet obvious to me what the right approach is or how much code reuse is possible without a lot of up-front engineering effort. You'll have an easier time getting started with some of the issues labeled "help wanted" and "good first issue."

qinqian commented 1 year ago

Got it, I'll review the issues labeled with "help wanted" or "good first issue" to see if I can volunteer.

KevinTran-TRI commented 4 months ago

Are there any updates on prioritization for this topic? Just curious.

eb8680 commented 4 months ago

Are there any updates on prioritization for this topic?

I'd say we're unlikely to get to it any time soon, though that could change if we took on a project that required it or a contributor wanted to own it. I also think it would require some substantial changes to the structure of the codebase to avoid effectively doubling our maintenance burden.