extra credit Search 85,062 technologies ⌘K Download for Mac

jaxtyping

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays.

Category
Python packages
pip install jaxtyping
↑↓ move ↵ open