贾克斯·普德

JAX 非常适合物理学，因为它可以通过数值求解器进行微分。本指南介绍了如何实现“优化友好”的传统 PDE 求解器以及如何构建神经混合物理模型。