jwave in n2cholas/awesome-jax

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.

8 +0

143 +0

21 +0

jaxdf in n2cholas/awesome-jax

A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations

created at Sept. 8, 2021, 4:38 p.m.

7 +0

122 +1

7 +0