wassname/jaxtyping
1
0
Fork 0
mirror of https://github.com/wassname/jaxtyping.git synced 2026-09-11 12:21:38 +08:00
Code Issues Packages Projects Releases Wiki Activity
83 Commits 2 Branches 31 Tags
v0.2.19
Commit Graph
9 Commits
Author SHA1 Message Date
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
Patrick Kidger e2f004afd4 Added support for jax.typing.ArrayLike; now works with PyTorch's bool 2023-02-25 17:01:27 -08:00
Patrick Kidger f30b7d1546 Update README.md 2023-01-20 07:44:45 -08:00
Patrick Kidger 2b1be5eb0a Update README.md 2022-12-07 17:22:57 -08:00
Patrick Kidger 8fbf7bf3a5 added link to eqxvision 2022-12-05 11:37:32 -08:00
Patrick Kidger 07e735797e Doc updates 2022-08-30 13:14:50 -07:00
Patrick Kidger 48e8131247 Added jaxtyping.Array=jnp.ndarray 2022-08-30 12:56:47 -07:00
Patrick Kidger 903000f3d5 Rewrote syntax 2022-08-30 12:47:45 -07:00
Patrick Kidger 7ac6ee04a8 initial commit 2022-07-01 21:01:46 +00:00