cupy decomposition jax machine-learning mxnet numpy python pytorch regression tensor tensor-algebra tensor-decomposition tensor-factorization tensor-learning tensor-methods tensor-regression tensorflow tensorly