TY - RPRT TI - Bringing PDEs to JAX with forward and reverse modes automatic differentiation AU - Ivan Yashchuk PY - 2023 UR - https://arxiv.org/abs/2309.07137 ID - 2309.07137 ER -