Jaxtyping | Anaconda.org

conda-forge / packages / jaxtyping 0.3.2

0

Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees.

copied from cf-post-staging / jaxtyping

To install this package run one of the following:
conda install conda-forge::jaxtyping