cutensor (pypi): NVIDIA cuTENSOR
Keywords: cuda|nvidia|machine|learning|tensor|network|high-performance|computing