/einops

Deep learning operations reinvented (for pytorch, tensorflow, jax and others)

Primary LanguagePythonMIT LicenseMIT

Watchers