New package py-tensorly (#37705)

This commit is contained in:
Benjamin Meyers 2023-05-18 16:38:23 -04:00 committed by GitHub
parent 3ff2fb69af
commit 641d4477d5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -0,0 +1,25 @@
# Copyright 2013-2023 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
from spack.package import *
class PyTensorly(PythonPackage):
"""TensorLy is a Python library that aims at making tensor learning simple
and accessible. It allows to easily perform tensor decomposition, tensor
learning and tensor algebra. Its backend system allows to seamlessly perform
computation with NumPy, PyTorch, JAX, MXNet, TensorFlow or CuPy, and run
methods at scale on CPU or GPU."""
homepage = "https://github.com/tensorly/tensorly"
pypi = "tensorly/tensorly-0.8.1.tar.gz"
maintainers("meyersbs")
version("0.8.1", sha256="cf78e4ffe612feca3510214002845c6831b267b1f2c1181154d41430310b237d")
depends_on("py-setuptools", type="build")
depends_on("py-numpy", type=("build", "run"))
depends_on("py-scipy", type=("build", "run"))