Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others) - View it on GitHub