davideger

I think your test is a great addition to the library independent of this fix. As a beginner at zig, I find it nice to have a small snippet of...

My understanding (and @patrick-kidger correct me if I am wrong) is that if you want jaxtyping to ensure that a return type dimension matches a scalar function parameter, you need...

I guess some context might help motivate this one. I was converting Sasha Rush's [tensor puzzles](https://github.com/srush/Tensor-Puzzles) from torchtyping to jaxtyping, and the final cell of his exercise is a competition...

Thanks for tracking down that this is not `jaxtyping` specific @knyazer, it sounds like IPython is the right place for this issue.