jaxtyping
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays.
- Category
- Python packages
pip install jaxtyping Official docs outline
jaxtyping.readthedocs.ioExtra Credit builds an outline of jaxtyping's docs the first time your work uses it. Until then, start at the official documentation.
More in Python packages
All 9,553-
azure-mgmt-batch Microsoft Azure Batch Management Client Library for Python
-
asyncstdlib The missing async toolbox
-
stamina Production-grade retries made easy.
-
safehttpx A small Python library created to help developers protect their applications from Server Side Request Forgery (SSRF) attacks.
-
reactivex ReactiveX (Rx) for Python
-
tableauserverclient A Python module for working with the Tableau Server REST API.
-
pyarrow-stubs Type annotations for pyarrow
-
multipart Parser for multipart/form-data
-
autoevals Universal library for evaluating AI models
-
simsimd Portable mixed-precision BLAS-like vector math library for x86 and ARM
-
janus Mixed sync-async queue to interoperate between asyncio tasks and classic threads
-
parsy Easy-to-use parser combinators, for parsing in pure Python