jaxtyping v0.2.17

(Yanked; had incompatibility with non-JAX installations. Prefer v0.2.19 instead.)

Full Changelog: v0.2.15...v0.2.17

jaxtyping v0.2.16

(Yanked; had static typing issues. Prefer v0.2.19 instead.)

Full Changelog: v0.2.15...v0.2.16

Jaxtyping v0.2.15

• Added support for functions in symbolic dimensions, e.g. "min(foo,bar)", which were previously disallowed due to the presence of a comma. (#51)
• Added support for adding ignored names to dimensions, e.g. "cols=4". (#76)
• Added support for Python 3.10 A | B union types (#73)
• Experimental (largely untested) compatibility with torch.compile. (#71)

Full Changelog: v0.2.14...v0.2.15

Jaxtyping v0.2.14

Highlights

• Silenced spurious pyright>=1.1.293 errors with jaxtyping.PyTree. (Thanks @ZacCranko in #66!)
• Compatibility with Plum v2 (beartype/plum#73).

Features

• Tweaked install_import_hook(..., typechecker=...) argument to also accept a string. In particular this means it can be used with beartype's new full checking; this can be enabled by passing typechecker="beartype.beartype(conf=beartype.BeartypeConf(strategy=beartype.BeartypeStrategy.On))".
• It is now the case that e.g. issubclass(Float[array_type, ...], array_type). (This is what brings compatibility with Plum v2.)
• NumPy and PyTorch compatibility is now tested.

Bugfixes

• Statements like Bool[int, "..."] should now raise an error.
• Applying @jaxtyping fo functions with fn in their __dict__ should now work.
• Subclassing dataclasses whilst using install_import_hook(..., typechecker="beartype.beartype") should no longer raise a spurious error.
• Changing the value of install_import_hook(..., typechecker=...) will no longer wrongly hit the same __pycache__. (Before this the change in value was ignored.)
• Can now serialise jaxtyping types using cloudpickle.

New Contributors

• @ZacCranko made their first contribution in #66

Full Changelog: v0.2.13...v0.2.14

Jaxtyping v0.2.13

Added support for jax.typing.ArrayLike.

Full Changelog: v0.2.12...v0.2.13

Jaxtyping v0.2.12

Autogenerated release notes as follows:

What's Changed

• Fix vanilla dataclasses by @brentyi in #56
• Fixes for some new failures. (Where did they come from?) by @patrick-kidger in #65

Full Changelog: v0.2.11...v0.2.12

Jaxtyping v0.2.11

Autogenerated release notes as follows:

What's Changed

• Hopefully fixed PyTree raising spurious errors. by @patrick-kidger in #54

Full Changelog: v0.2.10...v0.2.11

Jaxtyping v0.2.10

Autogenerated release notes as follows:

What's Changed

• Fixed working with the new (unreleased) version of typeguard by @patrick-kidger in #53

Full Changelog: v0.2.9...v0.2.10

Jaxtyping v0.2.9

Autogenerated release notes as follows:

What's Changed

• The import hook now decorates dataclass init methods by @patrick-kidger in #48
• JAX is no longer a hard dependency (to support e.g. PyTorch) by @patrick-kidger in #50
• Explicitly export names in jaxtyping.* by @brentyi in #49

New Contributors

• @brentyi made their first contribution in #49

Full Changelog: v0.2.8...v0.2.9

Jaxtyping v0.2.8

Autogenerated release notes as follows:

What's Changed

• NamedTuple example by @peterroelants in #36
• Silenced warning by @patrick-kidger in #40
• Update array_types.py by @patrick-kidger in #41
• update jaxtyped decorator by @murphyk in #44

New Contributors

• @peterroelants made their first contribution in #36
• @murphyk made their first contribution in #44

Full Changelog: v0.2.7...v0.2.8