V0.7.0: torch.compile, preserve axis identity, array api
Major changes:
torch.compile
just works, registration of operations happens automatically- JAX's distributed arrays can use ellipses, and in general ellipsis processing now preserves axis identity. This involved changing internal gears of einops.
- Array API:
einops
operations can be used with any framework that follows the standard (seeeinops.array_api
) - Python 3.7 is dead. Good bye, you were great at the time
- Gluon is dropped as previously announced
- reduce/repeat/rearrange all accept lists now
PRs list
- Preserve axis identity + drop python 3.7 by @arogozhnikov in #255
- Support array api by @arogozhnikov in #261
- add type-based caching by @arogozhnikov in #262
- Uniform support for list inputs across rearrange/reduce/repeat by @arogozhnikov in #263
- Drop gluon by @arogozhnikov in #264
- automatically register torch ops in torchdynamo by @arogozhnikov in #265
- release 0.7.0rc1, highlight changes in README by @arogozhnikov in #266
- cover dynamic shapes in torch.compile, introduce fallback if shape was not cacheable by @arogozhnikov in #275
- bump version by @arogozhnikov in #276
- fix #279, update description by @arogozhnikov in #281
- Add support for any/all reductions in einops.reduce by @arogozhnikov in #282
- maintenance: cache pip dependencies by @arogozhnikov in #283
- maintenance: update versions of github actions by @arogozhnikov in #285
Full Changelog: v0.6.1...v0.7.0