The Sequential Monte Carlo goes NUTS: boosting gravitational-wave inference

Sequential Monte Carlo (SMC) methods have recently been applied to gravitational-wave inference as a powerful alternative to standard sampling techniques, such as nested sampling. At the same time, gradient-based Markov Chain Monte Carlo algorithms, most notably the No-U-Turn Sampler (NUTS), provide an efficient way to explore high-dimensional parameter spaces. In this work we present , a Bayesian inference framework that combines the parallelism and evidence-estimation capabilities of SMC with the state-of-the-art sampling performance of NUTS. Moreover, exploits the local geometric structure of the posterior to further improve efficiency. Built on JAX, a high-performance computing framework that enables automatic differentiation and hardware acceleration, performs gravitational-wave inference on binary black-hole events in around ten minutes, yielding posterior samples and Bayesian evidence estimates that are consistent with those obtained through nested sampling. This work sets a new milestone in gravitational-wave inference with likelihood-based methods and paves the way for model comparison tasks to be accomplished in minutes.