From f30b7d1546dd9f3c9865b3b5fe0546cb5ddd792a Mon Sep 17 00:00:00 2001 From: Patrick Kidger <33688385+patrick-kidger@users.noreply.github.com> Date: Fri, 20 Jan 2023 07:44:45 -0800 Subject: [PATCH] Update README.md --- README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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