jaxdf (pypi): A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
Keywords: differentiable programming|differential equations|discretization|jax|pde|simulation|differentiable-programming