Jay Mody
Results
1
issues of
Jay Mody
Runtime type checking via `typeguard` causes `TypeError` due to array's having type `DeviceArray`.
3
I'm trying to use `jaxtyping` with runtime type checking via `typeguard` as described [`here`](https://github.com/google/jaxtyping/blob/dcd73e343124d1ed9d2bbd356a50fef6a60a45c9/API.md#option-1-jaxtypingjaxtyped). Here's my code: ```python import jax.numpy as jnp from jaxtyping import Array, Float, jaxtyped from typeguard...
question