jaxtyping¶
Type annotations and runtime type-checking for: 1. shape and dtype of JAX arrays; (Now also supports PyTorch, NumPy, and TensorFlow!) 2. PyTrees.
homepage: https://github.com/patrick-kidger/jaxtyping
| version | versionsuffix | toolchain |
|---|---|---|
0.2.38 |
-CUDA-12.6.0 |
foss/2024a |
(quick links: (all) - 0 - a - b - c - d - e - f - g - h - i - j - k - l - m - n - o - p - q - r - s - t - u - v - w - x - y - z)