Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Stars
2.6k
Forks
280
Watchers
2.6k
Open Issues
66
Overall repository health assessment
No package.json found
This might not be a Node.js project
Remove `vectorize_particles` from `RenyiELBO` docstring (Issue #2166) (#2168)
0152376View on GitHubExample: Embarrassingly Parallel MCMC via Consensus Monte Carlo (#2167)
878441dView on GitHubFix: Tracer leaks during NUTS sampling on random_nnx_module when NNX module contains mutable state (#2162)
d11c8dcView on GitHubfix: remove frozen project build during doc generation (#2160)
863f21cView on GitHubCI(gh-2126): transition to uv for dependency management and testing (#2129)
59f4452View on GitHubAlways call `jnp.finfo` on array `dtypes`, not arrays themselves (#2147)
af37655View on GitHub678
commits
228
commits
43
commits
42
commits
41
commits
33
commits
28
commits
19
commits
15
commits
12
commits