REVIEW 17 cited by
BlackJAX: Composable Bayesian inference in JAX
Not yet reviewed by Pith; the record is open.
This paper has not been read by Pith yet. Machine review is queued; the pith claim, tier, and objections will appear here once it completes.
SPECIMEN: schema-true, not a live event
T0 review · schema-true
One-sentence machine reading of the paper's core claim.
pith:XXXXXXXX · record.json · timestamp
read the original abstract
BlackJAX is a library implementing sampling and variational inference algorithms commonly used in Bayesian computation. It is designed for ease of use, speed, and modularity by taking a functional approach to the algorithms' implementation. BlackJAX is written in Python, using JAX to compile and run NumpPy-like samplers and variational methods on CPUs, GPUs, and TPUs. The library integrates well with probabilistic programming languages by working directly with the (un-normalized) target log density function. BlackJAX is intended as a collection of low-level, composable implementations of basic statistical 'atoms' that can be combined to perform well-defined Bayesian inference, but also provides high-level routines for ease of use. It is designed for users who need cutting-edge methods, researchers who want to create complex sampling methods, and people who want to learn how these work.
Forward citations
Cited by 17 Pith papers
-
On the triple nature of the PSR J0435+3233 system
PSR J0435+3233 is a hierarchical triple: its anomalous spin-down is an acceleration artifact from a ~1.2 M⊙ outer companion on a ~70-year eccentric orbit.
-
Numerical polology: towards next-generation model-building for cosmology
Numerical polology framework samples coupling space to discover ghost-free tensor field theories up to rank three for cosmology, then applies resulting priors to black hole superradiance, dynamical dark energy, and GW data.
-
The Orbital Eccentricity--Radius Distribution for Warm, Single Planets in TESS
Warm Jupiters are frequently eccentric (~59% in a high-eccentricity mode), while warm sub-Neptunes and sub-Saturns are mostly low-eccentricity, with the switch occurring at a break radius of about 9.8 Earth radii.
-
Higher-Order Hit-&-Run Samplers for Linearly Constrained Densities
New Hit-and-Run MCMC variants that incorporate gradient and curvature—LHR, smHR, smLHR—sample linearly constrained densities more efficiently than several existing samplers, with convergence guarantees.
-
Rapid Hubble constant inference from GW170817 using GPU-accelerated nested sampling: prior sensitivity and the limits of post-hoc reweighting
GPU-accelerated nested sampling on GW170817 demonstrates that switching to a uniform-in-dL prior shifts the H0 tail and median far more than post-hoc reweighting captures, due to an under-sampled (dL, iota) bimodality.
-
End-to-end differentiable retrieval of molecular spectra using hydrodynamics, chemistry, and radiative transfer
An end-to-end differentiable JAX pipeline couples 1D hydrodynamics, time-dependent chemistry, and radiative transfer, and recovers shock and rate parameters from synthetic HCO+ spectra.
-
Binary neutron stars in the next-generation era: Multi-messenger detection prospects and constraints on the equation of state, mass distribution, and cosmology
With ET (and ET+CE), mock multi-messenger BNS catalogues yield ~40–500 EM counterparts per year and, under ideal recovery, constrain R1.4 to ~0.2 km and H0 to ~1 km s−1 Mpc−1.
-
Ab Initio Real-Time Gravitational-Wave Parameter Estimation
Slice-within-Gibbs nested sampling on modern GPUs delivers well-calibrated BNS parameter estimation in ~12 minutes uncompressed and ~89 seconds with heterodyning, from cold priors.
-
Fast, accurate, and differentiable: a neural-network surrogate for NRSur7dq4 precessing binary black hole waveforms
A piecewise MLP surrogate emulates NRSur7dq4 over its full domain at NR-faithful accuracy with ~1 ms GPU latency and a fully differentiable JAX likelihood pipeline.
-
The Universal Warmup Path: Automatic Preconditioner Selection for HMC
A dimension-scheduled multi-chain controller automatically selects diagonal vs low-rank HMC preconditioners from warmup evidence and outperforms fixed Fisher and diagonal baselines on ESS per gradient.
-
The Sequential Monte Carlo goes NUTS: Boosting Gravitational-Wave Inference
SHARPy uses Sequential Monte Carlo with a No-U-Turn sampler in JAX to estimate gravitational-wave posteriors and evidence for binary black holes in about ten minutes.
-
Dynamic or Systematic? Bayesian model selection between dark energy and supernova biases
Bayesian evidence prefers a low-redshift supernova magnitude offset over dynamical dark energy when DES-5Y is combined with DESI BAO, but only under the assumption that Lambda CDM is correct.
-
Canonical Bayesian Linear System Identification
Bayesian inference for LTI systems in canonical state-space forms resolves parameter non-identifiability, yields well-behaved posteriors, and restores Bernstein-von Mises asymptotics.
-
Flow Annealed Importance Sampling Bootstrap meets Differentiable Particle Physics
Flow Annealed Importance Sampling Bootstrap, applied with replay buffering to differentiable matrix elements, achieves higher importance sampling efficiency per target evaluation than rKLD or fKLD in high-dimensional ...
-
KMT-2025-BLG-1314 and KMT-2025-BLG-1392: two microlensing planetary/brown-dwarf candidates analyzed with differentiable code
KMT-2025-BLG-1314 and KMT-2025-BLG-1392 are modeled as binary-lens microlensing events with planet/brown-dwarf candidates, and HMC with the differentiable microlux code is shown to handle bimodal posteriors better than emcee.
-
Gravitational-wave inference at GPU speed: A bilby-like nested sampling kernel within blackjax-ns
A GPU implementation of the bilby/dynesty acceptance-walk nested sampler recovers statistically equivalent posteriors and evidences with large core-hour speedups.
-
Bayesian Inverse Physics for Neuro-Symbolic Robot Learning
A position paper arguing that hybrid neuro-symbolic architectures combining physics, Bayesian inference, and program synthesis are essential for general-purpose robot learning.
Discussion (0). Continue with ORCID to comment.