Abstract:We propose the use of automatic differentiation through the programming framework JAX for accelerating a variety of analysis tasks throughout gravitational wave (GW) science. Firstly, we demonstrate that complete waveforms which cover the inspiral, merger, and ringdown of binary black holes (i.e., IMRPhenomD) can be written in JAX and demonstrate that the serial evaluation speed of the waveform (and its derivative) is similar to the lalsuite implementation in C . Moreover, JAX allows for graphics processing unit–accelerated waveform calls which can be over an order of magnitude faster than serial evaluation on a CPU. We then focus on three applications where efficient and differentiable waveforms are essential. Firstly, we demonstrate how gradient descent can be used to optimize the ∼200 coefficients that are used to calibrate the waveform model. In particular, we demonstrate that the typical match with numerical relativity waveforms can be improved by more than 50%. Secondly, we show that Fisher forecasting calculations can be sped up by ∼3–5× (on a CPU) with no loss in accuracy. This increased speed makes Fisher forecasting for a population of events substantially simpler. Finally, we show that gradient-based samplers like Hamiltonian Monte Carlo lead to significantly reduced autocorrelation values when compared to traditional Monte Carlo methods. Since differentiable waveforms have substantial advantages for a variety of tasks throughout GW science, we propose that waveform developers use JAX to build new waveforms moving forward. Our waveform code, ripple , can be found on GitHub website and will continue to be updated with new waveforms as they are implemented. https://doi.org/10.1103/PhysRevD.110.064028 Published by the American Physical Society under the terms of the Creative Commons Attribution 4.0 International license. Further distribution of this work must maintain attribution to the author(s) and the published article's title, journal citation, and DOI. Funded by Bibsam. Published by the American Physical Society

Recalibrating a gravitational wave phenomenological waveform model

Recalibrating Gravitational Wave Phenomenological Waveform Model

A phenomenological gravitational waveform model of binary black holes incorporating horizon fluxes

Enriching the Symphony of Gravitational Waves from Binary Black Holes by Tuning Higher Harmonics

PhenomXO4a: a phenomenological gravitational-wave model for precessing black-hole binaries with higher multipoles and asymmetries

Statistical Gravitational Waveform Models: What to Simulate Next?

Phenomenological gravitational-wave model for precessing black-hole binaries with higher multipoles and asymmetries

Waveform accuracy and systematic uncertainties in current gravitational wave observations

IMRPhenomXP_NRTidalv2: An improved frequency-domain precessing binary neutron star waveform model

Incorporating waveform calibration error in gravitational-wave modeling and inference for SEOBNRv4

Optimizing post-Newtonian parameters and fixing the BMS frame for numerical-relativity waveform hybridizations

Impact of numerical relativity information on effective-one-body waveform models

Survey of four precessing waveform models for binary black hole systems

Assessing the model waveform accuracy of gravitational waves

New and robust gravitational-waveform model for high-mass-ratio binary neutron star systems with dynamical tidal effects

Systematic bias from waveform modeling for binary black hole populations in next-generation gravitational wave detectors

Comparing gravitational waveform models for binary black hole mergers through a hypermodels approach

Accuracy limitations of existing numerical relativity waveforms on the data analysis of current and future ground-based detectors

Fast and faithful interpolation of numerical relativity surrogate waveforms using a meshfree approximation

Differentiable and hardware-accelerated waveforms for gravitational wave data analysis