/einops

Deep learning operations reinvented (supports tf, pytorch, chainer, gluon and others)

Primary LanguagePythonMIT LicenseMIT

Stargazers