diff --git a/README.md b/README.md index 6f9a517..6d3dbd9 100644 --- a/README.md +++ b/README.md @@ -2,9 +2,10 @@ Type annotations **and runtime checking** for: -1. shape and dtype of [JAX](https://github.com/google/jax) arrays; +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). + **For example:** ```python from jaxtyping import Array, Float, PyTree