GitHub repository: ucl-bug/jwave
Primary language: Python
Description: A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs