patrick-kidger/jaxtyping

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

语言：Python · GitHub 星标：1,879 · Fork：101

话题：jax · python-typing · typing

在 GitHub 上查看原仓库