link to why use type annotations

This commit is contained in:
Michael J Clark
2024-05-19 09:16:10 +08:00
committed by GitHub
parent 1c0186a1c9
commit 1456302503
+1 -1
View File
@@ -1,6 +1,6 @@
<h1 align="center">jaxtyping</h1>
Type annotations **and runtime type-checking** for:
[Use type annotations **and runtime type-checking**](https://jax.readthedocs.io/en/latest/jep/12049-type-annotations.html) for:
1. shape and dtype of [JAX](https://github.com/google/jax) arrays; *(Now also supports PyTorch, NumPy, and TensorFlow!)*
2. [PyTrees](https://jax.readthedocs.io/en/latest/pytrees.html).