typeddjango / awesome-python-typing

Collection of awesome Python types, stubs, plugins, and tools to work with them.

patrick-kidger / jaxtyping

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