Project Name | Stars | Downloads | Repos Using This | Packages Using This | Most Recent Commit | Total Releases | Latest Release | Open Issues | License | Language |
---|---|---|---|---|---|---|---|---|---|---|
Einops | 7,548 | 5 | 767 | 3 months ago | 14 | October 01, 2023 | 33 | mit | Python | |
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others) | ||||||||||
Cupy | 7,482 | 96 | 107 | 3 months ago | 138 | December 07, 2023 | 551 | mit | Python | |
NumPy & SciPy for GPU | ||||||||||
Tensorly | 1,466 | 10 | 39 | 4 months ago | 20 | March 08, 2023 | 64 | other | Python | |
TensorLy: Tensor Learning in Python. | ||||||||||
Speedtorch | 559 | 2 | 4 years ago | 14 | January 06, 2020 | 3 | mit | Python | ||
Library for faster pinned CPU <-> GPU transfer in Pytorch | ||||||||||
Pyinn | 248 | 6 years ago | 10 | mit | Python | |||||
CuPy fused PyTorch neural networks ops | ||||||||||
Pytorch Extension | 120 | a year ago | gpl-3.0 | Python | ||||||
an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors | ||||||||||
Structure Tensor | 45 | 4 months ago | 4 | June 07, 2022 | 2 | mit | Python | |||
Structure tensor 2D and 3D implementation for Python. | ||||||||||
Pytorch Custom C Cuda Operations | 8 | 4 years ago | Python | |||||||
Example of easily implementing custom tensor operations in C and CUDA. |