1.pytorch / pytorchTensors and Dynamic neural networks in Python with strong GPU accelerationneural-networkautogradgpunumpydeep-learningtensor+2 morePythonOther license105K total31K forks+6,186 starsfrom 98,884 · score 8.73
2.cupy / cupyNumPy & SciPy for GPUcudacudnncublascusolverncclpython+12 morePythonMIT12K total1.2K forks+1,513 starsfrom 10,877 · score 7.32
3.apache / tvmOpen Machine Learning Compiler Frameworkcompilertensordeep-learninggpuopenclmetal+7 morePythonAPACHE-2.014K total4K forks+623 starsfrom 13,263 · score 6.44