Add support for numpy struct dtype

patrick-kidger / jaxtyping

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

Other

1.05k stars 49 forks source link

Closed alexfanqi closed 1 month ago

alexfanqi commented 1 month ago

Fixes #204

alexfanqi commented 1 month ago

sorry, this took so long. How does this look?

patrick-kidger commented 1 month ago

LGTM! Thank you for the contribution :)