Follow
Bringing PDEs to JAX with forward and reverse modes automatic differentiation. (arXiv:2309.07137v1 [cs.MS]) http://arxiv.org/abs/2309.07137
Bringing PDEs to JAX with forward and reverse modes automatic differentiation. (arXiv:2309.07137v1 [cs.MS]) http://arxiv.org/abs/2309.07137