jaxtyping/docs/api/pytree.md at 0c596ff3739b7658912afaef2d0f8d4ed06f48ce - jaxtyping - Gitea: Git with a cup of tea

wassname/jaxtyping

mirror of https://github.com/wassname/jaxtyping.git synced 2026-08-29 11:20:25 +08:00

Files

T

Patrick Kidger 849b15db3b Added better docs; added PRNGKeys. Packaging with pyproject.toml.

Also added support for matching dtypes using regexes, and for nesting e.g. Shaped[Float[Array, "dim1 dim2"], "dim3"].

2023-05-10 15:27:25 -07:00

8 lines

166 B

Markdown

Raw Blame History

 # PyTree annotations
 :::jaxtyping.PyTree
     selection:
         members:
             false
 Note that `jaxtyping.PyTree` is only available if JAX has been installed.