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