patrick-kidger/jaxtyping v0.2.20

jaxtyping v0.2.20

on GitHub

• Added jaxtyping.PyTreeDef type.
• Can now detect x = jaxtyping.PyTree[foo] via issubclass(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