Architecture Acceleration of Machine Learning Gravitational Waveform Surrogate Models
Geophysics
This paper presents differentiable and hardware-accelerated implementations of the machine-learning surrogate models mlgw and mlgw-bns for gravitational-wave waveform modeling, enabling JIT compilation, vector parallelization, GPU acceleration, and automatic differentiation within JAX.
New See LICENSE file gravitational waveswaveform surrogate modelsGPU acceleration