ucl-bug / jwave

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Allow float64 for helmholtz solver

astanziola opened this issue · comments

For large domain and a lot of damping, the gradient calculation for the Helmholtz solver may be inaccurate using float32. It should be possible to ensure that all operations are done with float64