PyPI · #761 most downloaded on PyPI

einops

0.8.221M downloads/moarogozhnikov/einops ↗New releases in your daily digest, changelog included.

A new flavour of deep learning operations

Last release 3 months ago
05 Jul 2026
Release timing varies
gaps range from 5 weeks to 12 months
Most releases are documented
notes for 10 of 14 stable releases
1 version withdrawn
withdrawn after publishing
8 years old
18 releases · first in 2018

Release timeline

18 releases since 2018

20182020202220242026

Releases

0.9.0.dev05 Jul 2026pre-release
upgrade python versions, stop using/supporting python 3.9, set version to 0.9.0dev by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/396
What's Changed

upgrade python versions, stop using/supporting python 3.9, set version to 0.9.0dev by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/396

use list/dict/tuple instead of List/Dict/Tuple by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/397

typing.Sequence -> collections.abc.Sequence + switch off a couple of ruff rules by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/398

add einops playground by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/400

MLXBackend: Rename add_axes method to add_axis by @Datamance in https://github.com/arogozhnikov/einops/pull/401

framework name: mlx -> mlx.core by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/403

fix the einops.ReductionCallable type by @jorenham in https://github.com/arogozhnikov/einops/pull/405

modernize annotations with PEP 604 union syntax by @jorenham in https://github.com/arogozhnikov/einops/pull/410

explicitly mark type aliases as TypeAlias (PEP 613) by @jorenham in https://github.com/arogozhnikov/einops/pull/411

Add ruff rules, set correct requires-version, update ruff by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/412

proper typing of array_api submodule by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/413

remove unnecessary overloads by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/414

fix mypy errors in einops.parsing and add missing annotations by @jorenham in https://github.com/arogozhnikov/einops/pull/417

fill in the missing annotations in einops.layers.oneflow by @jorenham in https://github.com/arogozhnikov/einops/pull/421

fix mypy errors in einops.layers.paddle and add missing annotations by @jorenham in https://github.com/arogozhnikov/einops/pull/422

fix mypy errors in einops.layers.flax and add missing annotations by @jorenham in https://github.com/arogozhnikov/einops/pull/420

fix mypy errors in einops.layers.torch and add missing annotations by @jorenham in https://github.com/arogozhnikov/einops/pull/425

annotate the return type of array_api.asnumpy by @jorenham in https://github.com/arogozhnikov/einops/pull/418

fix mypy errors in einops.layers.__init__ and add missing annotations by @jorenham in https://github.com/arogozhnikov/einops/pull/419

add hatch run check by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/426

minor: remove unused function, adjust test by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/427

suppress/fix various mypy complains by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/428

delete fake setup.py by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/429

more typing work by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/430

resolve rest of mypy complains by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/432

create empty env "pypi" by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/437

switch deployment to uv from hatch by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/438

New Contributors

@Datamance made their first contribution in https://github.com/arogozhnikov/einops/pull/401

@jorenham made their first contribution in https://github.com/arogozhnikov/einops/pull/405

Full Changelog: https://github.com/arogozhnikov/einops/compare/v0.8.2...v0.9.0dev

Additional notes

What's Changed

upgrade python versions, stop using/supporting python 3.9, set version to 0.9.0dev by @arogozhnikov in #396
use list/dict/tuple instead of List/Dict/Tuple by @arogozhnikov in #397
typing.Sequence -> collections.abc.Sequence + switch off a couple of ruff rules by @arogozhnikov in #398
add einops playground by @arogozhnikov in #400
MLXBackend: Rename add_axes method to add_axis by @Datamance in #401
framework name: mlx -> mlx.core by @arogozhnikov in #403
fix the einops.ReductionCallable type by @jorenham in #405
modernize annotations with PEP 604 union syntax by @jorenham in #410
explicitly mark type aliases as TypeAlias (PEP 613) by @jorenham in #411
Add ruff rules, set correct requires-version, update ruff by @arogozhnikov in #412
proper typing of array_api submodule by @arogozhnikov in #413
remove unnecessary overloads by @arogozhnikov in #414
fix mypy errors in einops.parsing and add missing annotations by @jorenham in #417
fill in the missing annotations in einops.layers.oneflow by @jorenham in #421
fix mypy errors in einops.layers.paddle and add missing annotations by @jorenham in #422
fix mypy errors in einops.layers.flax and add missing annotations by @jorenham in #420
fix mypy errors in einops.layers.torch and add missing annotations by @jorenham in #425
annotate the return type of array_api.asnumpy by @jorenham in #418
fix mypy errors in einops.layers.__init__ and add missing annotations by @jorenham in #419
add hatch run check by @arogozhnikov in #426
minor: remove unused function, adjust test by @arogozhnikov in #427
suppress/fix various mypy complains by @arogozhnikov in #428
delete fake setup.py by @arogozhnikov in #429
more typing work by @arogozhnikov in #430
resolve rest of mypy complains by @arogozhnikov in #432
create empty env "pypi" by @arogozhnikov in #437
switch deployment to uv from hatch by @arogozhnikov in #438

New Contributors

@Datamance made their first contribution in #401
@jorenham made their first contribution in #405

Full Changelog: v0.8.2...v0.9.0dev

Open source →

0.8.226 Jan 2026

torch.compile: previously we relied on registration of einops in torch.compile. This registration was called by torch. Now with torch >= 2.8 torch.com

torch.compile: previously we relied on registration of einops in torch.compile. This registration was called by torch. Now with torch >= 2.8 torch.compile engine can compile einops operations without additional hints/registrations.
recommended backend for MLX is still einops.array_api.<operation>, but since community asked multiple times for support of einops.einsum, there is now MLX backend in einops, so einops.<operation> would also work with MLX.
minor improvements to docs, and mention of several related projects.

What's Changed

Update README.md by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/366
Update codebase, introduce consistency checks, update EinMix tutorial by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/374
Update python versions in tests, change min supported version to 3.9 by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/364
mention cubed as one more backend via array api by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/375
Extend tests for torch.compile by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/382
add einops.jl and einops for R in docs_src/pages/projects.md by @Qile0317 in https://github.com/arogozhnikov/einops/pull/383
mention ein color and AlphaFold3 by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/389
A bit more of compile tests by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/390
Patch #391 (conflicts in multithreading initialization) by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/392
switch off debug importing by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/393
Skip allow_in_graph if torch >= 2.8 by @guilhermeleobas in https://github.com/arogozhnikov/einops/pull/394
Add mlx backend by @ifsheldon in https://github.com/arogozhnikov/einops/pull/376
update testing by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/395

New Contributors

@Qile0317 made their first contribution in https://github.com/arogozhnikov/einops/pull/383
@guilhermeleobas made their first contribution in https://github.com/arogozhnikov/einops/pull/394
@ifsheldon made their first contribution in https://github.com/arogozhnikov/einops/pull/376

Full Changelog: https://github.com/arogozhnikov/einops/compare/v0.8.1...v0.8.2

Open source →

Additional notes

torch.compile: previously we relied on registration of einops in torch.compile. This registration was called by torch. Now with torch >= 2.8 torch.compile engine can compile einops operations without additional hints/registrations.
recommended backend for MLX is still einops.array_api.<operation>, but since community asked multiple times for support of einops.einsum, there is now MLX backend in einops, so einops.<operation> would also work with MLX.
minor improvements to docs, and mention of several related projects.

What's Changed

Update README.md by @arogozhnikov in #366
Update codebase, introduce consistency checks, update EinMix tutorial by @arogozhnikov in #374
Update python versions in tests, change min supported version to 3.9 by @arogozhnikov in #364
mention cubed as one more backend via array api by @arogozhnikov in #375
Extend tests for torch.compile by @arogozhnikov in #382
add einops.jl and einops for R in docs_src/pages/projects.md by @Qile0317 in #383
mention ein color and AlphaFold3 by @arogozhnikov in #389
A bit more of compile tests by @arogozhnikov in #390
Patch #391 (conflicts in multithreading initialization) by @arogozhnikov in #392
switch off debug importing by @arogozhnikov in #393
Skip allow_in_graph if torch >= 2.8 by @guilhermeleobas in #394
Add mlx backend by @ifsheldon in #376
update testing by @arogozhnikov in #395

New Contributors

@Qile0317 made their first contribution in #383
@guilhermeleobas made their first contribution in #394
@ifsheldon made their first contribution in #376

Full Changelog: v0.8.1...v0.8.2

0.8.19 Feb 2025

niceties: citation, docs, fixed broken links

What's Changed

TLDR:

ellipsis is added to EinMix
tests moved into the package
devcontainer provided
added backend for pyTensor
niceties: citation, docs, fixed broken links
this did not require any changes in einops, but array API is supported by more libs these days, and einops can operate on them

PRs:

Update README.md by @project-delphi in https://github.com/arogozhnikov/einops/pull/316
change tests for numpy2.0 + fix for numpy regression by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/325
PR from @pzread: fix tinygrad concat + changed formatting to fix CI errors by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/329
Add devcontainer and dockerfile by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/330
delete experimental/indexing by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/333
Use same ruff version in CI and in pre-commit, reformat notebooks by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/336
Delete chainer backend by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/335
Mention support of cupy, pydata/sparse, ndonnx and others by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/337
Moving tests into source tree by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/338
Update documentation, mention changes in testing + devcontainers by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/339
Auto-deploy mkdocs + update mkdocs requirements by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/341
Add dict syntax to the einops layers example by @eadadi in https://github.com/arogozhnikov/einops/pull/342
include CITATION.cff for einops by @luke-carlson in https://github.com/arogozhnikov/einops/pull/345
mention that cupy and jax now support array api by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/348
Fix docs broken links after master -> main migration by @VladKha in https://github.com/arogozhnikov/einops/pull/349
[Paddle] Update _backends.py to support pir.Value by @HydrogenSulfate in https://github.com/arogozhnikov/einops/pull/353
Update docstring for einops.reduce by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/358
add a comment about axes names by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/359
Add support of ellipsis to einmix by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/360
Add PyTensor backend by @ricardoV94 in https://github.com/arogozhnikov/einops/pull/362
rename input_dict, document parameter by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/363

New Contributors

@project-delphi made their first contribution in https://github.com/arogozhnikov/einops/pull/316
@eadadi made their first contribution in https://github.com/arogozhnikov/einops/pull/342
@luke-carlson made their first contribution in https://github.com/arogozhnikov/einops/pull/345
@VladKha made their first contribution in https://github.com/arogozhnikov/einops/pull/349
@HydrogenSulfate made their first contribution in https://github.com/arogozhnikov/einops/pull/353
@ricardoV94 made their first contribution in https://github.com/arogozhnikov/einops/pull/362

Full Changelog: https://github.com/arogozhnikov/einops/compare/v0.8.0...v0.8.1

Open source →

Additional notes

What's Changed

TLDR:

ellipsis is added to EinMix
tests moved into the package
devcontainer provided
added backend for pyTensor
niceties: citation, docs, fixed broken links
this did not require any changes in einops, but array API is supported by more libs these days, and einops can operate on them

PRs:

Update README.md by @project-delphi in #316
change tests for numpy2.0 + fix for numpy regression by @arogozhnikov in #325
PR from @pzread: fix tinygrad concat + changed formatting to fix CI errors by @arogozhnikov in #329
Add devcontainer and dockerfile by @arogozhnikov in #330
delete experimental/indexing by @arogozhnikov in #333
Use same ruff version in CI and in pre-commit, reformat notebooks by @arogozhnikov in #336
Delete chainer backend by @arogozhnikov in #335
Mention support of cupy, pydata/sparse, ndonnx and others by @arogozhnikov in #337
Moving tests into source tree by @arogozhnikov in #338
Update documentation, mention changes in testing + devcontainers by @arogozhnikov in #339
Auto-deploy mkdocs + update mkdocs requirements by @arogozhnikov in #341
Add dict syntax to the einops layers example by @eadadi in #342
include CITATION.cff for einops by @luke-carlson in #345
mention that cupy and jax now support array api by @arogozhnikov in #348
Fix docs broken links after master -> main migration by @VladKha in #349
[Paddle] Update _backends.py to support pir.Value by @HydrogenSulfate in #353
Update docstring for einops.reduce by @arogozhnikov in #358
add a comment about axes names by @arogozhnikov in #359
Add support of ellipsis to einmix by @arogozhnikov in #360
Add PyTensor backend by @ricardoV94 in #362
rename input_dict, document parameter by @arogozhnikov in #363

New Contributors

@project-delphi made their first contribution in #316
@eadadi made their first contribution in #342
@luke-carlson made their first contribution in #345
@VladKha made their first contribution in #349
@HydrogenSulfate made their first contribution in #353
@ricardoV94 made their first contribution in #362

Full Changelog: v0.8.0...v0.8.1

0.8.028 Apr 2024

resolve warning in py3.11 related to docstring

TLDR

tinygrad backend added
resolve warning in py3.11 related to docstring
remove graph break for unpack
breaking TF layers were updated to follow new instructions, new layers compatible with TF 2.16, and not compatible with old TF (certainly does not work with TF2.13)

What's Changed

Fix invalid escape sequence in einsum docstring by @atwam in https://github.com/arogozhnikov/einops/pull/298
Tinygrad support by @blueridanus in https://github.com/arogozhnikov/einops/pull/297
Coerce bool to int in unpack by @drubinstein in https://github.com/arogozhnikov/einops/pull/287
Remove oneflow from testing by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/289
tests: fix torch installation to force CPU by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/288
Allow anonymous axes in parse_shape, fix #302 by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/303
Codebase standards + update TF layers by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/318
update github actions by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/319

New Contributors

@drubinstein made their first contribution in https://github.com/arogozhnikov/einops/pull/287
@atwam made their first contribution in https://github.com/arogozhnikov/einops/pull/298
@blueridanus made their first contribution in https://github.com/arogozhnikov/einops/pull/297

Full Changelog: https://github.com/arogozhnikov/einops/compare/v0.7.0...v0.8.0

Open source →

Additional notes

TLDR

tinygrad backend added
resolve warning in py3.11 related to docstring
remove graph break for unpack
breaking TF layers were updated to follow new instructions, new layers compatible with TF 2.16, and not compatible with old TF (certainly does not work with TF2.13)

What's Changed

Fix invalid escape sequence in einsum docstring by @atwam in #298
Tinygrad support by @blueridanus in #297
Coerce bool to int in unpack by @drubinstein in #287
Remove oneflow from testing by @arogozhnikov in #289
tests: fix torch installation to force CPU by @arogozhnikov in #288
Allow anonymous axes in parse_shape, fix #302 by @arogozhnikov in #303
Codebase standards + update TF layers by @arogozhnikov in #318
update github actions by @arogozhnikov in #319

New Contributors

@drubinstein made their first contribution in #287
@atwam made their first contribution in #298
@blueridanus made their first contribution in #297

Full Changelog: v0.7.0...v0.8.0

Open source →

0.7.01 Oct 2023

torch.compile just works, registration of operations happens automatically

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 (see einops.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 https://github.com/arogozhnikov/einops/pull/255
Support array api by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/261
add type-based caching by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/262
Uniform support for list inputs across rearrange/reduce/repeat by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/263
Drop gluon by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/264
automatically register torch ops in torchdynamo by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/265
release 0.7.0rc1, highlight changes in README by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/266
cover dynamic shapes in torch.compile, introduce fallback if shape was not cacheable by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/275
bump version by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/276
fix #279, update description by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/281
Add support for any/all reductions in einops.reduce by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/282
maintenance: cache pip dependencies by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/283
maintenance: update versions of github actions by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/285

Full Changelog: https://github.com/arogozhnikov/einops/compare/v0.6.1...v0.7.0

Open source →

Additional notes

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 (see einops.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

0.7.0rc18 Jul 2023pre-release

torch.compile just works, registration of operations happens automatically

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 (see einops.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

What's Changed

Support array api by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/261
add type-based caching by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/262
Uniform support for list inputs across rearrange/reduce/repeat by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/263
Drop gluon by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/264
automatically register torch ops in torchdynamo by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/265
release 0.7.0rc1, highlight changes in README by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/266
Preserve axis identity + drop python 3.7 by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/255

Full Changelog: https://github.com/arogozhnikov/einops/compare/v0.6.1...v0.7.0rc1

Open source →

Additional notes

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 (see einops.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

What's Changed

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
Preserve axis identity + drop python 3.7 by @arogozhnikov in #255

Full Changelog: v0.6.1...v0.7.0rc1

Open source →

0.6.2rc05 Jul 2023pre-releasewithdrawn: 0.6.2 cancelled in favor of 0.7.0

pre-release is published to allow public testing of new caching logic (pattern analysis is now dependent on input dimensionality to preserve axis iden

pre-release is published to allow public testing of new caching logic (pattern analysis is now dependent on input dimensionality to preserve axis identity).

What's Changed

Preserve axis identity + drop python 3.7 by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/255

Full Changelog: https://github.com/arogozhnikov/einops/compare/v0.6.1...v0.6.2rc0

Open source →

Additional notes

pre-release is published to allow public testing of new caching logic (pattern analysis is now dependent on input dimensionality to preserve axis identity).

What's Changed

Preserve axis identity + drop python 3.7 by @arogozhnikov in #255

Full Changelog: v0.6.1...v0.6.2rc0

Open source →

0.6.119 Apr 2023

einops layers perfectly interplay with torch.compile

einops layers perfectly interplay with torch.compile
einops operations needs registration: run einops._torch_specific.allow_ops_in_compiled_graph() before torch.compile
paddle is now supported (thanks to @zhouwei25)
as previously announced, support of mxnet is dropped

What's Changed

Add PaddlePaddle backend for einops by @zhouwei25 in https://github.com/arogozhnikov/einops/pull/242 , https://github.com/arogozhnikov/einops/pull/245
Add allow_ops_in_compiled_graph to support torch.compile by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/251
torch.concat -> torch.cat to support packing for torch < 1.10 by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/238
chores: improve documentation and testing by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/246
Remove mxnet by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/231

New Contributors

@zhouwei25 made their first contribution in https://github.com/arogozhnikov/einops/pull/242

Full Changelog: https://github.com/arogozhnikov/einops/compare/v0.6.0...v0.6.1

Open source →

Additional notes

einops layers perfectly interplay with torch.compile
einops operations needs registration: run einops._torch_specific.allow_ops_in_compiled_graph() before torch.compile
paddle is now supported (thanks to @zhouwei25)
as previously announced, support of mxnet is dropped

What's Changed

Add PaddlePaddle backend for einops by @zhouwei25 in #242 , #245
Add allow_ops_in_compiled_graph to support torch.compile by @arogozhnikov in #251
torch.concat -> torch.cat to support packing for torch < 1.10 by @arogozhnikov in #238
chores: improve documentation and testing by @arogozhnikov in #246
Remove mxnet by @arogozhnikov in #231

New Contributors

@zhouwei25 made their first contribution in #242

Full Changelog: v0.6.0...v0.6.1

Open source →

0.6.09 Nov 2022Deprecated

Sunsetting experimental mxnet support: no demand and package is outdated, with numerous deprecations and poor support of corner cases. 0.6.0 will be t…

What's Changed

Introduce einops.pack and einops.unpack by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/222
Update example to match description by @EPronovost in https://github.com/arogozhnikov/einops/pull/217
Improve type hinting by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/221
Cosmetics for pack/unpack: documentation and comments by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/223
Preparations for 0.6.0 release by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/224

New Contributors

@EPronovost made their first contribution in https://github.com/arogozhnikov/einops/pull/217

Announcement

Sunsetting experimental mxnet support: no demand and package is outdated, with numerous deprecations and poor support of corner cases. 0.6.0 will be the last release with mxnet backend.

Full Changelog: https://github.com/arogozhnikov/einops/compare/v0.5.0...v0.6.0

Open source →

Additional notes

What's Changed

Introduce einops.pack and einops.unpack by @arogozhnikov in #222
Update example to match description by @EPronovost in #217
Improve type hinting by @arogozhnikov in #221
Cosmetics for pack/unpack: documentation and comments by @arogozhnikov in #223
Preparations for 0.6.0 release by @arogozhnikov in #224

New Contributors

@EPronovost made their first contribution in #217

Announcement

Sunsetting experimental mxnet support: no demand and package is outdated, with numerous deprecations and poor support of corner cases. 0.6.0 will be the last release with mxnet backend.

Full Changelog: v0.5.0...v0.6.0

Open source →

0.5.03 Oct 2022

Create einsum operation by @MilesCranmer in https://github.com/arogozhnikov/einops/pull/197

What's Changed

Create einsum operation by @MilesCranmer in https://github.com/arogozhnikov/einops/pull/197
Add flax layers by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/214
Add oneflow backend by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/181
Add oneflow backend by @rentainhe in https://github.com/arogozhnikov/einops/pull/180
Fix wrong error message by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/196
Clarify documentation re. default bias on EinMix by @maxeonyx in https://github.com/arogozhnikov/einops/pull/201
corrected spelling mistake: einsops -> einops by @cs-mshah in https://github.com/arogozhnikov/einops/pull/205
add mean-reduction for bfloat16, fix #206 by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/209
add py.typed (adopt PEP 561) by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/211
Delete tensorflow-specific readme file by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/212
Adopt pypa/hatch by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/213

New Contributors

@rentainhe made their first contribution in https://github.com/arogozhnikov/einops/pull/180
@MilesCranmer made their first contribution in https://github.com/arogozhnikov/einops/pull/197
@maxeonyx made their first contribution in https://github.com/arogozhnikov/einops/pull/201
@cs-mshah made their first contribution in https://github.com/arogozhnikov/einops/pull/205

Full Changelog: https://github.com/arogozhnikov/einops/compare/v0.4.1...v0.5.0

Open source →

0.4.018 Jan 2022

torch.jit.script is supported (in addition to previous torch.jit.trace)

Main Changes

torch.jit.script is supported (in addition to previous torch.jit.trace)
EinMix (swiss-knife for next-gen MLPs) is added. A much-improved einsum/linear layer is now available.
einops.repeat in torch does not create copy when possible

Detailed PRs

Update documentation by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/137
Multiple updates in docs, add Rearrange layer to torch test by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/138
Add support for torch scripting of einops layers by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/139
Introduce EinMix - swiss-knife for next-gen MLPs by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/142
Docs improvements: wording, visual style, EinMix by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/143
Move docs to a separate folder by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/144
Type hinting + add testing for EinMix composition/decomposition by @arogozhnikov in https://github.com/arogozhnikov/einops/pull/154
Reject repeated axes in parse_shape by @dmitriy-serdyuk in https://github.com/arogozhnikov/einops/pull/159
Enable ellipsis in patterns for parse_shape. by @dmitriy-serdyuk in https://github.com/arogozhnikov/einops/pull/162

New Contributors

@dmitriy-serdyuk made their first contribution in https://github.com/arogozhnikov/einops/pull/159

Full Changelog: https://github.com/arogozhnikov/einops/compare/v0.3.2...v0.4.0

Open source →