UCL Discovery
UCL home » Library Services » Electronic resources » UCL Discovery

j-Wave: An open-source differentiable wave simulator

Stanziola, A; Arridge, SR; Cox, BT; Treeby, BE; (2023) j-Wave: An open-source differentiable wave simulator. SoftwareX , 22 , Article 101338. 10.1016/j.softx.2023.101338. Green open access

[thumbnail of 1-s2.0-S2352711023000341-main.pdf]
Preview
Text
1-s2.0-S2352711023000341-main.pdf - Published Version

Download (1MB) | Preview

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.

Type: Article
Title: j-Wave: An open-source differentiable wave simulator
Open access status: An open access version is available from UCL Discovery
DOI: 10.1016/j.softx.2023.101338
Publisher version: https://doi.org/10.1016/j.softx.2023.101338
Language: English
Additional information: © 2023 The Authors. Published by Elsevier B.V. This is an open access article under the CC BY license (http://creativecommons.org/licenses/by/4.0/).
Keywords: Differentiable simulator, Acoustics, Machine learning, GPU acceleration, Wave equation, Helmholtz equation, JAX
UCL classification: UCL
UCL > Provost and Vice Provost Offices > UCL BEAMS
UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Engineering Science
UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Engineering Science > Dept of Computer Science
UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Engineering Science > Dept of Med Phys and Biomedical Eng
URI: https://discovery.ucl.ac.uk/id/eprint/10165839
Downloads since deposit
99Downloads
Download activity - last month
Download activity - last 12 months
Downloads by country - last 12 months

Archive Staff Only

View Item View Item