Wave Equation Python Packages

k-wave-python

A Python interface to k-Wave GPU accelerated binaries

jwave

A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs

wavetorch

🌊 Numerically solving and backpropagating through the wave equation