Pinned Loading

1. pyro-ppl/numpyro pyro-ppl/numpyro Public

   Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

   Python 2.6k 271

2. stan-dev/math stan-dev/math Public

   The Stan Math Library is a C++ template library for automatic differentiation of any order using forward, reverse, and mixed modes. It includes a range of built-in functions for probabilistic model...

   C++ 808 198

3. sbijax sbijax Public

   Simulation-based inference in JAX

   Python 44 5

4. ramsey ramsey Public

   Probabilistic deep learning using JAX

   Python 15 2

5. surjectors surjectors Public

   Surjection layers for density estimation with normalizing flows

   Python 4 3

6. blaxbird blaxbird Public

   A high-level API to build and train NNX models.

   Python 2