finitediffx (pypi): Finite difference tools in JAX.
Keywords: python|jax