jaxdf
A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
119 stars
7 watching
7 forks
Language: Python
last commit: 12 days ago
Linked from 1 awesome list
differentiable-programmingdiscretizationjaxpde
n2cholas/awesome-jax