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

Mayawaves: Python Library for Interacting with the Einstein Toolkit and the MAYA Catalog

Python Open Source Waveform Extractor (POWER): An open source, Python package to monitor and post-process numerical relativity simulations

Second MAYA Catalog of Binary Black Hole Numerical Relativity Waveforms

Initial Data and Eccentricity Reduction Toolkit for Binary Black Hole Numerical Relativity Waveforms

Mahakala: a Python-based Modular Ray-tracing and Radiative Transfer Algorithm for Curved Space-times

The GeoClaw software for depth-averaged flows with adaptive refinement

A package to bridge experimental tokamak data to modelling workflows for heating and transport

VAMPyR -- A High-Level Python Library for Mathematical Operations in a Multiwavelets Representation

Houdini for Astrophysical Visualization

Differentiable and hardware-accelerated waveforms for gravitational wave data analysis

VAMPyR —A high-level Python library for mathematical operations in a multiwavelet representation

Caustics: A Python Package for Accelerated Strong Gravitational Lensing Simulations

AMEP: The Active Matter Evaluation Package for Python

MATILDA.FT, a Mesoscale Simulation Package for Inhomogeneous Soft Matter

Modules for Experiments in Stellar Astrophysics (MESA): Convective Boundaries, Element Diffusion, and Massive Star Explosions

ExaGRyPE: Numerical General Relativity Solvers Based upon the Hyperbolic PDEs Solver Engine ExaHyPE

wfl Python Toolkit for Creating Machine Learning Interatomic Potentials and Related Atomistic Simulation Workflows

MPI-AMRVAC 3.0: updates to an open-source simulation framework

Tleco: A Toolkit for Modeling Radiative Signatures from Relativistic Outflows

pygwb: Python-based library for gravitational-wave background searches

IB2d Reloaded: a more powerful Python and MATLAB implementation of the immersed boundary method