Use jax.Array and jax.typing.ArrayLike for types

pnkraemer / probfindiff

Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and with out-of-the-box uncertainty quantification.

https://probfindiff.readthedocs.io

MIT License

10 stars 2 forks source link

Use jax.Array and jax.typing.ArrayLike for types #55

Closed pnkraemer closed 1 year ago