funsor (pypi): A tensor-like library for functions and distributions
Keywords: probabilistic|machine|learning|bayesian|statistics|pytorch|jax|machine-learning|numpy|probabilistic-programming|pyro|symbolic