PyPI page
Home page
Author:
Uber AI Labs
Summary:
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Latest version:
0.22.0
Required dependencies:
jax
|
jaxlib
|
multipledispatch
|
numpy
|
tqdm
Optional dependencies:
jax
Downloads last day:
11,524
Downloads last week:
173,892
Downloads last month:
960,818
7-day smoothing