teneva-jax (pypi): Compact implementation of basic operations in the tensor-train (TT) format with jax framework for approximation, optimization and sampling with multidimensional arrays and multivariate functions
Keywords: low-rank|representation|tensor|train|format|TT-decomposition|cross|approximation|als|anova|jax