jwave

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

GitHub
137 stars
8 watching
21 forks
Language: Python
last commit: 12 days ago
Linked from 1 awesome list

acousticsdifferentiable-simulationsgpugpu-accelerationjaxkwavephysics-informed-neural-networksscientific-machine-learningsimulationtpu-accelerationultrasoundwave-equation

Backlinks from these awesome lists: