We should have our own array abstraction that can run with pytorch/numpy/jax as backend + handles indexing the way we do it

kach / memo

A domain-specific probabilistic programming language for reasoning about reasoning

https://pypi.org/project/memo-lang/

37 stars 2 forks source link

Closed kach closed 2 months ago

kach commented 4 months ago

possibly just numpy+jax, see #24

kach commented 2 months ago

i think we have now committed to jax