A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs
created at April 6, 2021, 10:22 a.m.
A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
created at Sept. 8, 2021, 4:38 p.m.