A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs
simulation gpu gpu-acceleration ultrasound wave-equation acoustics jax tpu-acceleration scientific-machine-learning differentiable-simulations physics-informed-neural-networks kwave
-
Updated
Sep 17, 2024 - Python