Einops v0.4.0: EinMix and torch.jit.script

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

New Contributors

• @dmitriy-serdyuk made their first contribution in #159

Full Changelog: v0.3.2...v0.4.0