Update README.md

This commit is contained in:
Patrick Kidger
2022-12-07 17:22:57 -08:00
committed by GitHub
parent 7b3d9a2e9a
commit 2b1be5eb0a
-2
View File
@@ -28,8 +28,6 @@ def accepts_pytree_of_arrays(x: PyTree[Float[Array, "batch c1 c2"]]):
pip install jaxtyping
```
Requires JAX 0.3.4+.
Also install your favourite runtime type-checking package. The two most popular are [typeguard](https://github.com/agronholm/typeguard) (which exhaustively checks every argument) and [beartype](https://github.com/beartype/beartype) (which checks random pieces of arguments).
## Documentation