github patrick-kidger/jaxtyping v0.3.5
jaxtyping v0.3.5

4 days ago
  • Feature: support TypeAliasTypes as array types. For example:

    type Foo = np.ndarray | int
    Float[Foo, ""]

    In particular, this re-enables support for numpy.typing.ArrayLike, which recently switched from being a union to a type alias.

  • Docs: big update to emphasize the way in which we are framework-agnostic.

Full Changelog: v0.3.4...v0.3.5

Don't miss a new jaxtyping release

NewReleases is sending notifications on new releases.