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
82 Commits 2 Branches 31 Tags
v0.2.18
Commit Graph
4 Commits
Author SHA1 Message Date
Patrick Kidger 0c596ff373 Fix for non-JAX installations. 2023-05-10 17:16:42 -07:00
Patrick Kidger 7934d2afed Static typing fixes 2023-05-10 16:35:45 -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
Patrick Kidger f0b240df5f Switched to ruff 2023-03-15 22:36:24 -07:00