arogozhnikov/einops: Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
Category: ml-dl
Language: Python
Topics: chainer, cupy, deep-learning, einops, jax, keras, numpy, pytorch, tensor, tensorflow
Homepage: https://einops.rocks