Finally, we're open sourced our 2D spectral Navier-Stokes solver with high-order time-stepping in JAX-CFD: github.com/google/jax-cfd…
It's differentiable and fast on GPU/TPU, so we hope it may be a useful reference for future ML for PDE efforts, e.g., like twitter.com/unsorsodicorda… |
No comments:
Post a Comment