py-jaxtyping

Pydantic support for Jaxtyping array annotations


Install
pip install py-jaxtyping==0.1.4