is there any way to use jaxtyping but not other type checks?

[galmacky]

Hi,

I'm trying to adopt jaxtyping for my pytorch project (quite large) using beartype. But it enforces so many other type checks and I cannot track down all the other errors caused by others as that would be too much work. Is there any way to only enforce jaxtyping type checks?

Many thanks!

[patrick-kidger]

Perhaps implement a typechecker=... of your own, that implements just the checks you want?