wassname/jaxtyping
1
0
Fork 0
mirror of https://github.com/wassname/jaxtyping.git synced 2026-08-29 11:20:25 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
10e1852b379ef97004d2f80df7d57e64aaed7127
jaxtyping/docs/api/pytree.md
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

166 B
Raw Blame History

PyTree annotations

:::jaxtyping.PyTree selection: members: false

Note that jaxtyping.PyTree is only available if JAX has been installed.

Reference in New Issue View Git Blame Copy Permalink