j-Wave: An open-source differentiable wave simulator
This provides a customizable and reusable tool for researchers and practitioners in acoustics and machine learning, though it is incremental as it builds on existing simulation concepts with added differentiability.
The authors introduced j-Wave, an open-source differentiable acoustic simulator that solves time-varying and harmonic acoustic problems, and validated its accuracy against established tools like k-Wave and other software.
We present an open-source differentiable acoustic simulator, j-Wave, which can solve time-varying and time-harmonic acoustic problems. It supports automatic differentiation, which is a program transformation technique that has many applications, especially in machine learning and scientific computing. j-Wave is composed of modular components that can be easily customized and reused. At the same time, it is compatible with some of the most popular machine learning libraries, such as JAX and TensorFlow. The accuracy of the simulation results for known configurations is evaluated against the widely used k-Wave toolbox and a cohort of acoustic simulation software. j-Wave is available from https://github.com/ucl-bug/jwave.