36 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Simulation Based Inference of Cosmological parameters in Jax using type Ia super... | 10K | |
| Parametric modeling in JAX + Equinox | 7K | |
| Parametric radio frequency modeling, optimization and sampling | 4K | |
| State of the art inference for your bayesian models. | 2K | |
| probabilistic programming focused on fun | 2K | |
| An (experimental) "unified" interface for probabilistic inference in JAX + Equin... | 2K | |
| Simulation-based inference in JAX | 2K | |
| A research-focused modular generative modeling library built on JAX/Flax NNX | 1K | |
| Continuous-discrete dynamical systems with JAX and related libraries. | 1K | |
| NumPyro integration for dynamical systems | 1K | |
| A probabilistic programming framework | 1K | |
| Opifex Computational Framework - JAX-native platform for scientific machine lear... | 903 | |
| Fast, lightweight and parallelised simulation-based inference in JAX. | 894 | |
| Tools for evolutionary forecasting. | 718 | |
| A Bayesian Spectral Line Modeling Framework | 679 | |
| Bayesian Multistate Bennett Acceptance Ratio (BayesMBAR) methods for free energy... | 520 | |
| Sequential Monte Carlo and particle filtering in JAX | 519 | |
| CPU-optimized JAX-based homodyne scattering analysis for XPCS under nonequilibri... | 493 | |
| Bayesian linear regression with variational inference, inspired by R's lm() and ... | 383 | |
| Regularized Stein thinning using JAX | 352 | |
| A package that contains a set of models for making football prediction. | 318 | |
| A time-series forecasting package with an intuitive API capable of modeling shor... | 312 | |
| Probabilistic methods of parameter inference for ordinary differential equations | 285 | |
| Differentiable models of enzyme-catalysed reaction networks | 284 | |
| Bayesian inference of population size history from recombining sequence data. | 258 | |
| Bayesian Data Analysis using PyMC and JAX | 255 | |
| :open_hands: Probabilistic reconciliation of time series forecasts | 253 | |
| Training Neural Network potentials through customizable routines in JAX. | 225 | |
| Efficient library for spectral analysis in high-energy astrophysics. | 220 | |
| Bayesian statistical framework for reconstructing past Earth system change from ... | 195 | |
| A Gibbs sampler for timing gamma-ray pulsars | 152 | |
| A JAX-based gaussian process structure learning package for time series modellin... | 129 | |
| Faster Hamiltonian trajectories for problems with guesses, using Blackjax | 114 | |
| Bayesian Deep Ensembles via MILE: easy to use, scikit-learn compatible and fast ... | 114 | |
| Jax library for probabilistic computations | 76 | |
| Implementation of Orbital Torus Imaging, and tools for pipelining runs. | 60 |