jaxtyping
Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees.
Installation
To install this package, run one of the following:
Usage Tracking
0.3.7
0.3.6
0.3.5
0.3.4
0.3.2
5 / 8 versions selected
Downloads (Last 6 months): 0
About
Summary
Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees.
Last Updated
Jan 31, 2026 at 00:32
License
MIT
Supported Platforms
noarch