{"library":"jaxtyping","install":[{"cmd":"pip install jaxtyping","imports":["from jaxtyping import Array","from jaxtyping import Float","from jaxtyping import Int","from jaxtyping import set_array_typecheck_enabled"]}]}