pymcx (pypi): Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.
Keywords: probabilistic-programming