Export Array from jax without installation about jaxtyping HOT 2 CLOSED

So jaxtyping should already support being used in non-JAX projects. For example:

from torch import Tensor
from jaxtyping import Float

def f(x: Float[Tensor, "foo bar"]): ...

Moreover jaxtyping does not even list JAX as a dependency.

from jaxtyping.

Agh, so sorry, I totally missed the interchangeability of Tensor and ndarray etc in the annotation. I was trying to import Array, which then gave a jax import error, under the mistaken impression that the sense in which the library is interoperable was that Array could be used to hint for both numpy, torch, etc.

Thanks so much.

from jaxtyping.