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