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