wassname/jaxtyping
1
0
Fork 0
mirror of https://github.com/wassname/jaxtyping.git synced 2026-09-08 17:10:08 +08:00
Code Issues Packages Projects Releases Wiki Activity
125 Commits 2 Branches 31 Tags
9e1ba8a77da38ab7e77a5663bf2f0e87f74b4a94
Commit Graph
3 Commits
Author SHA1 Message Date
Patrick Kidger e55348a4b4 Added jaxtyping.Real 2023-10-17 09:33:03 -07:00
Patrick Kidger e308695293 Updated to support both new and old style JAX PRNG keys, as they are going to co-exist simultaneously. See https://github.com/google/jax/pull/17297 2023-09-14 20:00:46 -07:00
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