jaxtyping v0.2.20
- Added
jaxtyping.PyTreeDef
type. - Can now detect
x = jaxtyping.PyTree[foo]
viaissubclass(x, jaxtyping.PyTree)
. - Fixed #89, in which
__builtins__
was getting added as an extra key to the memo stack. - Renamed modules with a leading underscore to indicate that they're private.
- Bump minimum Python version to 3.9.
Full Changelog: v0.2.19...v0.2.20