jaxtyping v0.3.2

https://github.com/patrick-kidger/jaxtyping/releases/tag/v0.3.2