Einsum Is All You Need: NumPy, PyTorch and TensorFlow