GitHub repository: rlouf/mcx
Primary language: Python
Description: Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.