A Python interface to k-Wave GPU accelerated binaries
A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs
Numerically solving and backpropagating through the wave equation using pytorch