Annotating tensor shapes using Python types
Why do you think that https://github.com/patrick-kidger/jaxtyping is a good alternative to tensor_annotations
Annotating tensor shapes using Python types
Why do you think that https://github.com/patrick-kidger/jaxtyping is a good alternative to tensor_annotations