V0.6.1: support paddle, support pytorch.compile
- einops layers perfectly interplay with torch.compile
- einops operations needs registration: run
einops._torch_specific.allow_ops_in_compiled_graph()
beforetorch.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