j-Wave: An open-source differentiable wave simulator
Computational Physics
2022-07-05 v1 Machine Learning
Mathematical Software
Sound
Audio and Speech Processing
Medical Physics
Abstract
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.
Keywords
Cite
@article{arxiv.2207.01499,
title = {j-Wave: An open-source differentiable wave simulator},
author = {Antonio Stanziola and Simon R. Arridge and Ben T. Cox and Bradley E. Treeby},
journal= {arXiv preprint arXiv:2207.01499},
year = {2022}
}