22 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Distrax: Probability distributions in JAX. | 206K | |
| Meridian is an MMM framework that enables advertisers to set up and run their ow... | 83K | |
| TF-Agents: A reliable, scalable and easy to use TensorFlow library for Contextua... | 58K | |
| Probabilistic Programming and Nested sampling in JAX | 23K | |
| SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms | 3K | |
| Surjection layers for density estimation with normalizing flows | 2K | |
| Simulation-based inference in JAX | 2K | |
| NumPyro integration for dynamical systems | 1K | |
| A probabilistic programming framework | 1K | |
| Opifex Computational Framework - JAX-native platform for scientific machine lear... | 903 | |
| GEMlib scientific compute library for epidemic modelling | 738 | |
| IEEE 802.11 MAPC (C-SR) simulator | 641 | |
| margarine: Posterior Sampling and Marginal Bayesian Statistics | 591 | |
| A JAX-based statistical sampling toolkit | 456 | |
| IEEE 802.11 DCF Simulator | 366 | |
| Modular JAX-based toolbox for implementing RL algorithms. | 305 | |
| :open_hands: Probabilistic reconciliation of time series forecasts | 253 | |
| Gaussian process with spherical harmonic features in JAX | 164 | |
| stan2tfp is a lightweight interface to the TensorFlow Probability backend of the... | 135 | |
| Deep Bayesian unsupervised decoder networks. Use poisson or multinomial belief n... | 106 | |
| Implementation of the PROTECT method | 69 | |
| Implementation of the PROTECT method for proxy-based causal inference | 67 |